项目需要tensorflow-gpu2.xx版本,而之前装的cuda9.0与cudnn与tensorflow-gpu2.xx不兼容,tensorflow-gpu2.3.0和cuda10可以匹配,tensorflow-gpu2.4.0就需要cuda11以上。
一、cuda9.0卸载
将cuda9.0的插件卸载,cudnn7.2.1不需要额外操作。
二、安装cuda10.1 和cudnn7.6.5
双击安装即可
全选即可
之前重装系统时将VC卸载,此时提示我未找到合适的VC版本,下载后继续安装即可
cudnn下的文件分别移入cuda根目录下对应的文件中即可
添加环境变量
C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.1\lib\x64
C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.1\bin
C:\ProgramData\NVIDIA Corporation\CUDA Samples\v10.1\common\lib\x64
C:\ProgramData\NVIDIA Corporation\CUDA Samples\v10.1\bin\win64
安装成功
三、安装tensorflow-gpu 2.3.0
卸载之前装的版本,pip install 安装即可,推荐阿里云的镜像,蹭蹭快。
搞定!