목록GPU (3)

Azure Zest

[ python ] GPU error ( InvalidArgumentError )

주피터 노트북으로 딥러닝 코드를 돌리다가 다음과 같은 오류가 발생하였다. InvalidArgumentError: device CUDA:4 not supported by XLA service while setting up XLA_GPU_JIT device number 4 처음 보는 오류였고, 이 오류로 인해 코드가 다 돌아가지 않았다. 혹시 GPU 메모리가 부족해서 나타난 오류인가 싶어 다음과 같이 사용하는 GPU를 바꾸어 주었더니 오류가 해결되었다. import os os.environ['CUDA_VISIBLE_DEVICES'] = '5' GPU 할당하는 것은 다음 링크에서 더 자세히 알아볼 수 있다. 윈도우에서 GPU 할당하기 [ Python ] 윈도우에서 GPU 할당하기 Linux에서 GPU 할당하기..
Deep Learning 2020. 5. 9. 23:26
[ Python ] 윈도우에서 GPU 할당하기

Linux에서 GPU 할당하기 리눅스에서 GPU를 정하는 것은 쉽다. 코드를 돌릴 때, 앞에 CUDA_VISIBLE_DEVICES 를 붙이기만 하면된다. 아래와 같이 숫자는 각각 0부터 하나씩 지정이 되어 있을테니, 숫자를 바꿔서 각각 GPU에서 다른 코드를 돌려볼 수 있다. $ CUDA_VISIBLE_DEICES=0 python main.py Windows에서 GPU 할당하기 기본적으로 윈도우에서는 리눅스 명령어가 먹히지 않으므로, 저 명령어가 먹히지 않는다. 따라서 실행시킬 코드 위에 아래 코드를 적어 놓는다. 당연히, 리눅스에서도 아래의 코드를 미리 적어놓으면 앞에 CUDA_VISIBLE_DEVICES를 붙이지 않고, 바로 GPU를 할당할 수 있다. 위와 마찬가지로 숫자를 다르게 써서, 선택할 수 ..
Deep Learning 2019. 7. 18. 09:02
[ Tensorflow ] GPU Ubuntu 16.04에 Anaconda 및 Tensorflow 설치하기

1. Anaconda 설치하기 먼저 아래의 링크를 타고, anaconda 설치 스크립트 파일을 다운받는다. 현재 버전에 해당하는 아나콘다를 받을 시에는 위의 링크로, 기존의 다른 버전의 아나콘다를 다운받을 시에는 밑의 링크를 타면 된다.(파이썬 3.6을 받으려면 Anaconda3-5.2.0이 제일 알맞은 것 같다.) 현재 아나콘다 version 다운받기 Anaconda Python/R Distribution - Anaconda The open-source Anaconda Distribution is the easiest way to perform Python/R data science and machine learning on Linux, Windows, and Mac OS X. With over 11 ..
Deep Learning 2019. 7. 1. 15:32
이전 Prev 1 Next 다음