Tensorflow에서 GPU를 사용하기 위해서는 다음이 필요하다
Pytorch로 이미 GPU를 이용하고 있는 딥러닝 연구자라면 1.과 2.는 이미 설치되어 있을 것이다. 하지만 이 경우에도 3. CuDNN 설치는 별도로 필요하다.
conda install -c anaconda cudnn
Reference