1:首先查看自己电脑的NVIDIA版本,我的是10.0所以CUda也安装10.0
2:去Cuda官网安装自己电脑的版本:
下载完以后运行软件程序
之后按照软件要求安装即可
添加环境变量
C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v9.0\libnvvp
C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v9.0\bin
C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v9.0\lib
C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v9.0\include
之后打开命令窗输入nvcc -V,显示版本号则成功
3安装cudnn,同样去官网下载,只是需要注册账号才能下载
4按照cudnn、cuda、tensorflow的版本号关系,如下所示
对应的更改tensorflow的版本号
我把我原先装的tensorflow2.2卸载了,然后重装了2.0.0版本。
安装过程如果过慢可以看我过往的博客。
随后pycharm运行原来写好的程序,可以发现 此时还会报些莫名其妙的错。比如找不到库文件、或者是检测不到tensorflow等。我检查了很久,最后尝试关闭了pycharm重启电脑。
再次打开pycharm,就能跑的很快了。
使用cpu速度
使用GPU速度
目测差别十几倍的样子