Programming/Deep Learning

[Tesorflow] GPU 설정 버전

BadaGreen_Kim 2020. 12. 1. 12:00

Tensorflw gpu 버전을 설치할때 Cuda 와 Cudnn 에 대해 버전별로 맞추어야하는 상황이 자주발생하는데 Tensorflw 버전마다 아래와 같이 Cuda와 CuDnn 을 설치했던 내용을 정리한다. 

 

  1. Cuda 10.1 + CuDNN 7.6.5 (normally > 7.6) + TensorFlow 2.2.0 (TF >= 2.1 requires CUDA >=10.1)
  2. Cuda 10.1 + CuDNN 7.6.5 (normally > 7.6) + TensorFlow 2.1.0 (TF >= 2.1 requires CUDA >= 10.1)
  3. Cuda 10.0 + CuDNN 7.6.3 + / TensorFlow 1.13/1.14 / TensorFlow 2.0.
  4. Cuda 9.0 + CuDNN 7.0.5 + TensorFlow 1.10 works