在安装TensorFlow-GPU时需要对应TensorFlow-GPU版本与CUDA、CUDNN的版本对应。每次需要更改环境的时候,都要各种找网址。这里做一个网址整理:
1、安装GPU版本之前先检查电脑是否支持GPU版本
链接: https://pan.baidu.com/s/19Sf1WUizsMDhGm4ax86hWA?pwd=df12https://pan.baidu.com/s/19Sf1WUizsMDhGm4ax86hWA?pwd=df12 提取码: df12
双击exe,安装软件选择否。
1代表的是显卡驱动版本,2代表的是支持CUDA计算
2、查找要安装的gpu版本和对应cuda、cudnn之间的版本对应关系
https://www.tensorflow.org/install/source_windows3、查看电脑显卡驱动版本是否符合对应CUDA版本要求
除了前面使用TechPowerUp GPU-Z 软件查看外还可以使用: cmd输入nvidia-smi
nvidia-smi显示不是内部或外部命令也不是可运行的程序
链接:Release Notes :: CUDA Toolkit Documentation
4、下载对应版本的CUDA
CUDA Toolkit Archive | NVIDIA Developer
题外话,下载太慢可以安装一个IDM
链接: https://pan.baidu.com/s/1pFEYiwEMFHvyNPjttpDH_g?pwd=m35d https://pan.baidu.com/s/1pFEYiwEMFHvyNPjttpDH_g?pwd=m35d 提取码: m35d
5、下载对应版本的CUDNN
cuDNN Archive | NVIDIA Developer(需要注册账号)
6、安装CUDA和CUDNN
Cuda和cuDNN安装教程(超级详细)_jhsignal的博客-CSDN博客_cudnn安装