목록tensorflow gpu 사용 시 (1)

로봇이 되고픈 부엉이

(tensorflow) GPU 여러개를 나누어 tensorflow 돌릴 때 쓰는 코드

python라이브러리 GPU-tensorflow를 설치하신 여러분들이 이 글을 많이 볼 것이라고 생각이 됩니다. 바로 해결책을 알려드리자면, Linux 리눅스 에서 CUDA_VISIBLE_DEVICES = 0 위 숫자는 GPU에 붙은 숫자인데, 숫자가 반대일 수 있고 아닐 수도 있다. 그러나 GPU 2개를 사용하고 있다면, 그냥 0 또는 1 두 개로 part allocation(부분 할당) 하여 사용할 수 있으니 그냥 0과 1만 잘 기억하여 사용하자. Windows 윈도우 에서 import os os.environ["CUDA_VISIBLE_DEVICES"] = "0" 윈도우는 리눅스와 유사하다. 그러나 main 함수 안에 적는다. 그럼 알아서 잘 작동한다. 유의사항은 위 리눅스와 유사하다. GPU가 2개..
Deep Learning/python 파이썬 Tip 2019. 7. 23. 15:45
이전 Prev 1 Next 다음