Tensorflow 와 CUDA, CUDNN 호환 버전 정리
CUDA & cuDNN
tensorflow-gpu 버전을 사용하기 위해서는 CUDA와 cuDNN 설치가 필요하다.
tensorflow-gpu 버전마다 설치해야 할 CUDA와 CUDNN 버전이 다르다.
tensorflow-gpu==1.5.0 버전부터는 CUDA 9와 cuDNN 7이 필요하다.
| TF 버전 | CUDA | cuDNN |
|---|---|---|
| TF 1.8 | 9 | 7 |
| ~ | 9 | 7 |
| TF 1.5 | 9 | 7 |
| TF 1.4 | 8 | 6 |
또한, 그래픽 카드마다 설치가 가능한 CUDA와 cuDNN 버전이 다를 수 있다.
이를 알기 위해서는 NVIDIA GPU 의 compute compatibility 확인해야 한다.
https://developer.nvidia.com/cuda-gpus#collapse2
Tesla V100 7.0
Tesla P100 6.0
Tesla P40 6.1
Tesla K40 3.5
Tesla C2075 2.0
GRID K-Series 3.x
TITAN V 7.0
TITAN X, Xp 6.1
GTX 1080 Ti 6.1
GTX 1080 6.1
CUDA 버전
CUDA 설치 시 필요한 최소 그래픽 Driver 버전이 존재한다.
| CUDA 버전 | Driver |
|---|---|
| CUDA 9.2 | 396.xx |
| CUDA 9.1 | 387.xx |
| CUDA 9.0 | 384.xx |
| CUDA 8.0 (GA2) | 375.xx |
| CUDA 8.0 | 367.4x |
| CUDA 7.5 | 352.xx |
| CUDA 7.0 | 346.xx |
