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

其中CUDA的下载链接为:

CUDA Toolkit Archive | NVIDIA Developer

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

cuDNN Archive | NVIDIA Developer

GPU

VersionPython versioncuDNNCUDA
tensorflow_gpu-2.11.03.7-3.108.111.2
tensorflow_gpu-2.10.03.7-3.108.111.2
tensorflow_gpu-2.9.03.7-3.108.111.2
tensorflow_gpu-2.8.03.7-3.108.111.2
tensorflow_gpu-2.7.03.7-3.98.111.2
tensorflow_gpu-2.6.03.6-3.98.111.2
tensorflow_gpu-2.5.03.6-3.98.111.2
tensorflow_gpu-2.4.03.6-3.88.011.0
tensorflow_gpu-2.3.03.5-3.87.610.1
tensorflow_gpu-2.2.03.5-3.87.610.1
tensorflow_gpu-2.1.03.5-3.77.610.1
tensorflow_gpu-2.0.03.5-3.77.410
tensorflow_gpu-1.15.03.5-3.77.410
tensorflow_gpu-1.14.03.5-3.77.410
tensorflow_gpu-1.13.03.5-3.77.410
tensorflow_gpu-1.12.03.5-3.67.29.0
tensorflow_gpu-1.11.03.5-3.679
tensorflow_gpu-1.10.03.5-3.679
tensorflow_gpu-1.9.03.5-3.679
tensorflow_gpu-1.8.03.5-3.679
tensorflow_gpu-1.7.03.5-3.679
tensorflow_gpu-1.6.03.5-3.679
tensorflow_gpu-1.5.03.5-3.679
tensorflow_gpu-1.4.03.5-3.668
tensorflow_gpu-1.3.03.5-3.668
tensorflow_gpu-1.2.03.5-3.65.18
tensorflow_gpu-1.1.03.55.18
tensorflow_gpu-1.0.03.55.18

Logo

为开发者提供学习成长、分享交流、生态实践、资源工具等服务,帮助开发者快速成长。

更多推荐