(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개라면 0과 1만 사용하여 한다.