不同版本的Tensorflow需对应不同的CUDA和cudnn版本,否者容易安装失败。可按下图所示,根据想要安装的Tensorflow版本,选择对应版本的CUDA和cudnn。

其中CUDA的下载链接为:

CUDA Toolkit Archive | NVIDIA Developer

cudnn下载链接(需要登录开发者账号)为:

cuDNN Archive | NVIDIA Developer

GPU

Version Python version cuDNN CUDA
tensorflow_gpu-2.11.0 3.7-3.10 8.1 11.2
tensorflow_gpu-2.10.0 3.7-3.10 8.1 11.2
tensorflow_gpu-2.9.0 3.7-3.10 8.1 11.2
tensorflow_gpu-2.8.0 3.7-3.10 8.1 11.2
tensorflow_gpu-2.7.0 3.7-3.9 8.1 11.2
tensorflow_gpu-2.6.0 3.6-3.9 8.1 11.2
tensorflow_gpu-2.5.0 3.6-3.9 8.1 11.2
tensorflow_gpu-2.4.0 3.6-3.8 8.0 11.0
tensorflow_gpu-2.3.0 3.5-3.8 7.6 10.1
tensorflow_gpu-2.2.0 3.5-3.8 7.6 10.1
tensorflow_gpu-2.1.0 3.5-3.7 7.6 10.1
tensorflow_gpu-2.0.0 3.5-3.7 7.4 10
tensorflow_gpu-1.15.0 3.5-3.7 7.4 10
tensorflow_gpu-1.14.0 3.5-3.7 7.4 10
tensorflow_gpu-1.13.0 3.5-3.7 7.4 10
tensorflow_gpu-1.12.0 3.5-3.6 7.2 9.0
tensorflow_gpu-1.11.0 3.5-3.6 7 9
tensorflow_gpu-1.10.0 3.5-3.6 7 9
tensorflow_gpu-1.9.0 3.5-3.6 7 9
tensorflow_gpu-1.8.0 3.5-3.6 7 9
tensorflow_gpu-1.7.0 3.5-3.6 7 9
tensorflow_gpu-1.6.0 3.5-3.6 7 9
tensorflow_gpu-1.5.0 3.5-3.6 7 9
tensorflow_gpu-1.4.0 3.5-3.6 6 8
tensorflow_gpu-1.3.0 3.5-3.6 6 8
tensorflow_gpu-1.2.0 3.5-3.6 5.1 8
tensorflow_gpu-1.1.0 3.5 5.1 8
tensorflow_gpu-1.0.0 3.5 5.1 8

Logo

AtomGit 是由开放原子开源基金会联合 CSDN 等生态伙伴共同推出的新一代开源与人工智能协作平台。平台坚持“开放、中立、公益”的理念,把代码托管、模型共享、数据集托管、智能体开发体验和算力服务整合在一起,为开发者提供从开发、训练到部署的一站式体验。

更多推荐