How to use tensorflow GPU

참고

일단 해당 가상환경 CMD를 킨후에 pip install tensorflow 를 한다.

그후에 텐서플로우의 버전을 확인후 텐서플로우 공식홈페이지 가서 버전을 확인후

호환가능한 CUDA버전과 CUDNN버전을 다운받고 합친다.

필자는 텐서플로우 버전이 2.3.1이며 버전 확인은 아래와 같이 하면 된다.

버전 확인후 2.3.1과 맞는

CUDA 10.1을 다운받았으며 CUDNN도 7.4 버전을 다운받았다.

다운받은후 C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.1 폴더안에 CUDNN을 합쳤으며

제어판에서 GPU 정상 동작을 확인하였다.