1.前言

最近在学习图象分类,用到了tensorflow,安装过程让人头大,装好后总结了一下过程:

重点:安装GPU版本的tensorflow首先得明确版本对应。

最近几个版本的对应见下图:

看tensotflow 是不是gpu版本 tensorflow的gpu版本_cuda


2.安装tensorflow

权衡了一下,选择了2.3.0版本的tensorflow,直接用pip安装指定版本即可

看tensotflow 是不是gpu版本 tensorflow的gpu版本_tensorflow_02

pipi install tensorflow==2.3.0

安装成功:

看tensotflow 是不是gpu版本 tensorflow的gpu版本_tensorflow_03


等待完成即可。

验证安装是否成功:

在python里import tensorflow,若无报错,则说明安装成功。

3.安装cuda

我的tensorflow2.3.0对应cuda版本为10.1,下载地址:官网cuda下载

看tensotflow 是不是gpu版本 tensorflow的gpu版本_图像识别_04


可选本地和在线下载工具

看tensotflow 是不是gpu版本 tensorflow的gpu版本_cuda_05


(貌似在线下载工具会快很多)

打开下载的exe文件:

看tensotflow 是不是gpu版本 tensorflow的gpu版本_tensorflow_06


选择一个路径存放下载资源(并非最终cuda安装位置)

而后自动下载文件,下载完成后,安装界面:

看tensotflow 是不是gpu版本 tensorflow的gpu版本_tensorflow_07


选择精简模式即可,等待安装完成关闭即可。

看tensotflow 是不是gpu版本 tensorflow的gpu版本_图像识别_08


验证安装是否成功:

看tensotflow 是不是gpu版本 tensorflow的gpu版本_tensorflow_09

nvcc -V

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

4.cudnn安装:

1,选择和你下载的cuda相应版本的cudnn,(好像最新的cuda都有对应的cudnn版本)所以我在这里下了最新的。 (修改:经后期使用反馈,在这里最好安装推荐版本7.6.5,否则会出现一些文件缺失问题。)

下载地址:官网cudnn下载地址

看tensotflow 是不是gpu版本 tensorflow的gpu版本_tensorflow_10

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

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

下,替换掉它本身的这三个文件。

看tensotflow 是不是gpu版本 tensorflow的gpu版本_深度学习_11


3,最后来进行环境变量的配置:

右键此电脑->属性->高级系统设置->环境变量,在path里新建下面五个环境变量:

看tensotflow 是不是gpu版本 tensorflow的gpu版本_cuda_12


看tensotflow 是不是gpu版本 tensorflow的gpu版本_机器学习_13


配置完成后即可。

4,验证安装是否成功:

进入目录:C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.1\extras\demo_suite

看tensotflow 是不是gpu版本 tensorflow的gpu版本_图像识别_14


执行bandwidthTest.exe与deviceQuery.exe两个程序。

返回Result = PASS说明安装成功。

看tensotflow 是不是gpu版本 tensorflow的gpu版本_图像识别_15

5.补上:

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