Win10中TensorFlow的GPU环境的配置

GPU环境的配置

在深度学习中,模型往往很复杂,数据量很大,此时使用GPU运行会快很多。而使用GPU就需要安装CUDA和CuDNN。本文是为了使用tensorflow的gpu版本而使用的GPU。

(1)GPU的配置

首先,要使用TensorFlow的GPU,需要达到的硬件前提:显卡类型是NVIDIA,显卡的计算能力要至少达到3.0。
可以在下面的网站查看:
https://developer.nvidia.com/cuda-gpus

然后,下载CUDA和CuDNN
CuDA至少8.0,CuDNN至少6.1
CuDA的下载网址:
https://developer.nvidia.com/cuda-downloads?target_os=Windows&target_arch=x86_64&target_version=10&target_type=exelocal

CuDNN的下载网址:
https://developer.nvidia.com/rdp/cudnn-download
注:CuDNN版本和CUDA的版本要对应上
我使用的是CUDA 8.0和CuDnn v7
这里写图片描述

这里写图片描述

Cudnn在官网下载后,直接解压即可,当中会有如下文件:
这里写图片描述


主要要注意的是CUDA的安装:

双击安装的cuda_8.0.44.exe后,出现

这里写图片描述


类似上图的画面,这个是CUDA安装包提取的路径,感觉没有什么意义,而且最终cuda的安装路径也不一定是这里(为什么说不一定,请继续看下去)


点击“OK”之后,经过一个进度条,之后会对系统进行检测。


正式到了安装的时候,建议选择“高级 - 自定义”。之后,都使用默认的选项,一直下一步,最后直到安装完成。


如果选择的是“精装”,后面的安装过程直接就会把CUDA安装在上面的提取枯井中;

如果选择“自定义”,后面也是可以自定义路径的,但是这样的话,CUDA可能不会自动配置好系统路径。



当安装CUDA之后,可以在控制台中输入: nvcc -V查看cuda的版本,如下图:

这里写图片描述


同时,环境变量中会多出两个路径,即:CUDA_PATH和CUDA_PATH_V8.0
如下图所示:
这里写图片描述

之后,将解压后的Cudnn文件夹中的bin、include和lib文件夹复制到下图所示的文件夹中:

这里写图片描述


还要在PATH中,添加
这里写图片描述

至此,CUDA和CuDNN的配置已完成

(2)Python中TensorFlow-gpu的安装

本机的环境:Python 3.5

当然,最好还是使用Anconda(包含Python3.6),去安装TensorFlow。

安装TensorFlow的gpu版本,只需要体格命令:

pip install tensorflow-gpu

安装完成后,在控制台中输入python,进入python环境
输入:

import tensorflow as tf

结果显示,一大堆错误,形如:
这里写图片描述

这个主要错误,就是找不到cudnn64_6.dll
解决问题:在下载Cudnn的时候,下载v6版本的CuDnn;如果下载了比v6更新的版本的,直接在修改,如下图所示:
这里写图片描述

将cudnn64_7改为cudnn64_6即可,注意高版本的可以直接修改为低版本的,但是反之则不行!

此时,再次输入:

import tensorflow as tf

则没有报错

再对tf.Session()进行测试
输入:

sess = tf.Session()
a = tf.constant(2)
b = tf.constant(3)
print(sess.run(a+b))

TensorFlow会自己调用空闲的GPU,而且会显示GPU的信息
同时,可以输入

sess = tf.Session(config=tf.ConfigProto(log_device_placement=True))

这个可以查看自己显卡的计算能力,和TensorFlow要求的计算能力,如果达不到,则同样起不了GPU加速的作用。

猜你喜欢

转载自blog.csdn.net/louishao/article/details/78827691