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 |
Thank you for your sharing. I am worried that I lack creative ideas. It is your article that makes me full of hope. Thank you. But, I have a question, can you help me?
You have greatly impressed me! Up until now, I don’t believe I’ve ever read anything even remotely like to it. I am very happy to have found someone with some original thoughts on this topic. To tell the truth, I appreciate you starting this. The internet is searching for someone with a little bit of imagination, and this website is just what someone needs!