목록InvalidArgumentError (1)

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
이전 Prev 1 Next 다음