tensorflow-gpu==1.8.0 을 설치하여도 GPU가 아닌 CPU로 계산할 때, GPU로 계산하도록 하는 방법

tensorflow를 설치하면 cpu 연산을, tensorflow-gpu를 설치하면 gpu연산을 할 수 있다. 따라서 다음 명령어를 이용해서 tensorflow-gpu를 설치한다.

 

pip install --upgrade tensorflow-gpu

 

위의 명령어를 치면 GPU 연산이 가능해지나 tensorflow-gpu의 최신버전(현재는 tensorflow-gpu==1.13.0) 이 설치된다. 따라서 코드가 실행되도록 하기 위해 tensorflow-gpu를 downgrade 시킨다.

 

pip install tensorflow-gpu==1.8.0

 

 

만약 tensorflow-gpu를 설치해도 gpu연산이 되지 않으면 NVIDIA 드라이버가 지워진 것일 수도 있다. 컴퓨터를 재부팅해보고 다음 명령어를 terminal에 입력해본다.

 

nvidia-smi

 

드라이버 정보가 뜨지 않으면 NVIDIA 드라이버가 없는 것이므로 아래의 블로그를 참조하여 NVIDIA 드라이버를 다시 설치한다.

 

https://devyurim.github.io/development%20environment/ubuntu/2018/05/28/ubuntu-3.html