首先我的电脑显卡配置是rtx 2070super,相应的驱动程序版本为如下图所示(建议要手动去官网安装最新的几个版本的显卡驱动,如果找不到去问下买显卡的客服)
其中我所支持的cuda版本为(可以在nvidia控制面板左下角系统信息查看,如下图,就是NVCUDA DLL那一栏),我的显卡最高支持的cuda版本为11.0,因此只要cuda版本低于这个都可以安装,但是目前最新的windows cudnn仅支持到cuda10.1版本,因此必须下载下载cuda10.1版本,这样可以找到cudnn支持:
目前网上也很很多简便的安装方式,就是利用annaconda去安装,但是这种方式我安装失败了,因此只能老老实实的下载cuda 以及cudnn。这种方法肯定是可以的,因此建议不要偷懒,老老实实的去下载把。
这里面我主要参考这篇博客【TensorFlow2.1.0安装教程_weixin_44170512的博客-CSDN博客_tensorflow2.1安装
https://blog.csdn.net/weixin_44170512/article/details/103990592】
按照这篇博客上面的操作找到相应的cuda以及cudnn下载即可。
这里面有几个需要注意的地方!!!!
【1】在安装cuda时一定要按照链接的这篇博客要求把Visual Studio Intergration这个选项给去掉,否则会出现错误!!
【2】一定要安装visual studio,目前官网最新的是visual studio 2019,我们仅需要下载最新的2019免费版社区即可。这个软件很大,仅需要选择以下几个即可,如下图
下载完成之后也不用管他是干嘛的,打开一下然后退出即可。
【3】一定要配置cudnn的环境变量,并且要把这个环境变量放到第一个!!按照上面那篇博客操作即可。
可以参考以下博客的代码
【https://blog.csdn.net/lidichengfo0412/article/details/102637824】
如果输出为ture则说明安装成功
另外有些同学可能出现以下错误 :Could not load dynamic library ‘cudnn64_7.dll’; dlerror: cudnn64_7.dll not found。
这个就是上面说的cudnn的环境变量没有加进去,如果出现这个错误再手动加入一下
我自己的系统变量如下图所示,第一个就是cudnn的环境变量。