windows 10 tensorflow-gpu 2.2.0安装

记录一下windows安装tensorflow-gpu 2.2.0过程

查看自己的显卡驱动版本及支持的cuda版本

首先我的电脑显卡配置是rtx 2070super,相应的驱动程序版本为如下图所示(建议要手动去官网安装最新的几个版本的显卡驱动,如果找不到去问下买显卡的客服)windows 10 tensorflow-gpu 2.2.0安装_第1张图片
其中我所支持的cuda版本为(可以在nvidia控制面板左下角系统信息查看,如下图,就是NVCUDA DLL那一栏),我的显卡最高支持的cuda版本为11.0,因此只要cuda版本低于这个都可以安装,但是目前最新的windows cudnn仅支持到cuda10.1版本,因此必须下载下载cuda10.1版本,这样可以找到cudnn支持:
windows 10 tensorflow-gpu 2.2.0安装_第2张图片

下载cuda以及对应的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免费版社区即可。这个软件很大,仅需要选择以下几个即可,如下图
windows 10 tensorflow-gpu 2.2.0安装_第3张图片
下载完成之后也不用管他是干嘛的,打开一下然后退出即可。
【3】一定要配置cudnn的环境变量,并且要把这个环境变量放到第一个!!按照上面那篇博客操作即可。

测试tensorflow-gpu 2.2.0

可以参考以下博客的代码
【https://blog.csdn.net/lidichengfo0412/article/details/102637824】
如果输出为ture则说明安装成功
windows 10 tensorflow-gpu 2.2.0安装_第4张图片

另外有些同学可能出现以下错误 :Could not load dynamic library ‘cudnn64_7.dll’; dlerror: cudnn64_7.dll not found。
这个就是上面说的cudnn的环境变量没有加进去,如果出现这个错误再手动加入一下
我自己的系统变量如下图所示,第一个就是cudnn的环境变量。
windows 10 tensorflow-gpu 2.2.0安装_第5张图片

你可能感兴趣的:(python)