tensorflow安装GPU版本(含CUDA和CUDNN的安装)【windows】,超详细个人总结!

1.前言

最近在学习图象分类,用到了tensorflow,安装过程让人头大,装好后总结了一下过程:
重点:安装GPU版本的tensorflow首先得明确版本对应。
最近几个版本的对应见下图:tensorflow安装GPU版本(含CUDA和CUDNN的安装)【windows】,超详细个人总结!_第1张图片
图片来源:tensorflow官网
(若以后更新版本,可跳转自行查看)

2.安装tensorflow

权衡了一下,选择了2.3.0版本的tensorflow,直接用pip安装指定版本即可
在这里插入图片描述

pipi install tensorflow==2.3.0

安装成功:
tensorflow安装GPU版本(含CUDA和CUDNN的安装)【windows】,超详细个人总结!_第2张图片
等待完成即可。
验证安装是否成功:
在python里import tensorflow,若无报错,则说明安装成功。

3.安装cuda

我的tensorflow2.3.0对应cuda版本为10.1,下载地址:官网cuda下载
tensorflow安装GPU版本(含CUDA和CUDNN的安装)【windows】,超详细个人总结!_第3张图片
可选本地和在线下载工具
tensorflow安装GPU版本(含CUDA和CUDNN的安装)【windows】,超详细个人总结!_第4张图片
(貌似在线下载工具会快很多)
打开下载的exe文件:
tensorflow安装GPU版本(含CUDA和CUDNN的安装)【windows】,超详细个人总结!_第5张图片
选择一个路径存放下载资源(并非最终cuda安装位置)
而后自动下载文件,下载完成后,安装界面:
tensorflow安装GPU版本(含CUDA和CUDNN的安装)【windows】,超详细个人总结!_第6张图片
选择精简模式即可,等待安装完成关闭即可。
tensorflow安装GPU版本(含CUDA和CUDNN的安装)【windows】,超详细个人总结!_第7张图片
验证安装是否成功:
tensorflow安装GPU版本(含CUDA和CUDNN的安装)【windows】,超详细个人总结!_第8张图片

nvcc -V

安装好会显示如上版本信息。(若显示不是内部或外部命令,可尝试重启电脑)

4.cudnn安装:

1,选择和你下载的cuda相应版本的cudnn,(好像最新的cuda都有对应的cudnn版本)所以我在这里下了最新的。 (修改:经后期使用反馈,在这里最好安装推荐版本,否则会出现一些文件缺失问题。)
下载地址:官网cudnn下载地址
tensorflow安装GPU版本(含CUDA和CUDNN的安装)【windows】,超详细个人总结!_第9张图片

2,下载完成后将压缩包里的bin include lib三个文件夹复制到你的cuda根目录(默认在:

C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.1

下,替换掉它本身的这三个文件。
tensorflow安装GPU版本(含CUDA和CUDNN的安装)【windows】,超详细个人总结!_第10张图片
3,最后来进行环境变量的配置:
右键此电脑->属性->高级系统设置->环境变量,在path里新建下面五个环境变量:
tensorflow安装GPU版本(含CUDA和CUDNN的安装)【windows】,超详细个人总结!_第11张图片
tensorflow安装GPU版本(含CUDA和CUDNN的安装)【windows】,超详细个人总结!_第12张图片
配置完成后即可。
4,验证安装是否成功:
进入目录:C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.1\extras\demo_suite
在这里插入图片描述
执行bandwidthTest.exe与deviceQuery.exe两个程序。
返回Result = PASS说明安装成功。
tensorflow安装GPU版本(含CUDA和CUDNN的安装)【windows】,超详细个人总结!_第13张图片

5.补上:

由于下载的是cudnn最新的版本,导致在使用tensorflow过程中出现了not found cudnn_64.7.dll的问题,经检查发现新版本cudnn的bin目录下原本应为cudnn_64.7.dll文件被更名为cudnn_74.8.dll,手动改回后问题解决。

你可能感兴趣的:(深度学习,机器学习,深度学习,tensorflow,cuda,图像识别)