Azure Zest

주피터 노트북으로 딥러닝 코드를 돌리다가 다음과 같은 오류가 발생하였다. 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 할당하기..

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