1.配置环境:
(1)cuda_10.0.130_411.31_win10
(2)cudnn-10.0-windows10-x64-v7.3.1.20
(3)tensorflow_gpu-1.12.0-cp36-cp36m-win_amd64.whl
2.步骤:
(1)下载cuda_10.0.130_411.31_win10
(2)安装cuda_10.0.130_411.31_win10
(3)下载cudnn-10.0-windows10-x64-v7.3.1.20
(4)将cudnn-10.0-windows10-x64-v7.3.1.20压缩包解压得到以下文件
将所有的4个文件(全部)复制到,文件夹C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.0,(我这里安装的10.0版本的cuda,路径是默认的安装路径)
(5)设置环境变量
在用户变量中添加变量
查看系统变量中是否存在下面两个
(6)安装TensorFlow
安装的时候一定要写上版本。
注意事项:
1. cuda 和 cudann 和 TensorFlow的版本一定要对应好。三者之间都是相互依赖的关系,有一个错了,import都会出问题。
2. 当时自己安装TensorFlow时,没有写上版本号,安装到一半,给ctrl + c 进行终止了(默认安装最新的版本),后来又写上了版本号,虽然安装上了,但是 因为 TensorFlow自动安装的依赖包不同,所以最后出现了依赖包冲突的问题。使用的笨方法卸载Python重新安装的(进行卸载TensorFlow,再重新安装时不好使的偶)