Windows 7安装GPU版本tensorflow(CUDA 9.0+cudnn7.4.1)GPU(NVIDIA Quarter K2000)

由于学习需要,手上有一台图形工作站,GPU是NVIDIA Quarter K2000的,就试着看看装个GPU环境训练

1 安装前准备工作

在官网可以查到自己的GPU是否支持CUDA加速
一般算力在3.0的才可以支持cudnn加速,如果自己的显卡版本没在这上面,还是老老实实使用CPU训练吧(不过一般N卡好像都支持cuda)

2 所需的软件

  • CUDA Tookit 9.0
  • cuDNN 7.4.1
    注意:现在最新的CUDA是10.0,不过最好还是装9.0一下的吧,本人装10.0后会出现问题,另外,cuDnn的版本一定要跟CUDA的版本对应着,一定,一定。

3 下载安装

3.1 安装CUDA

Windows 7安装GPU版本tensorflow(CUDA 9.0+cudnn7.4.1)GPU(NVIDIA Quarter K2000)_第1张图片
官网根据个人电脑配置自行下载对应的CUDA Toolkit
下载完成后是个exe文件,双击执行就行了,跟普通软件安装方法一样,
中间选择的地方选择 精简安装

安装完成后

CUDA自动添加环境变量
在这里插入图片描述

查看是否安装成功

nvcc -V

出现如下结果表示安装成功
Windows 7安装GPU版本tensorflow(CUDA 9.0+cudnn7.4.1)GPU(NVIDIA Quarter K2000)_第2张图片
否则安装失败,重新卸载干净后安装

附:使用功能nvidia-smi查看自己的GPU

找到装好的cuda目录
将此路径加入系统变量Path中

C:\Program Files\NVIDIA Corporation\NVSMI

然后命令行输入nvidia-smi
可以查看自己的显卡信息
Windows 7安装GPU版本tensorflow(CUDA 9.0+cudnn7.4.1)GPU(NVIDIA Quarter K2000)_第3张图片

3.2 下载cuDNN并配置环境变量

在这里插入图片描述
解压下载好的cudnn文件,解压的目录可以随便设置,主要是为了将路径加入到环境变量中

D:\Program Files\Nvidia\cuda\bin

在这里插入图片描述
将此文件所在的路径加入到系统环境变量中Path
安装完成

4 在Anaconda建立Tensorflow-gpu虚拟环境

安装Anaconda就不在多说直接开始安装虚拟环境
打开anaconda prompt
输入

conda create --name tensorflow-gpu python=3.6 anaconda

解释
create --name tensorflow 创建tensorflow虚拟环境的名称是tensorflow-gpu
python=3.6 对应的Python版本号
anaconda 对应安装的anaconda的相关依赖库例如numpy,scikit-learn,pandas等等

5 安装Tensorflow GPU版本&Keras

安装tensorflow-gpu

pip install tensorflow-gpu

安装keras

pip install keras

安装完成

6 测试

打开anaconda prompt
输入

activate tensprflow-gpu 

激活tensorflow-gpu环境
Windows 7安装GPU版本tensorflow(CUDA 9.0+cudnn7.4.1)GPU(NVIDIA Quarter K2000)_第4张图片
另外安装完成后在开始菜单中会出现这样的图片
如图
Windows 7安装GPU版本tensorflow(CUDA 9.0+cudnn7.4.1)GPU(NVIDIA Quarter K2000)_第5张图片
输入python
Windows 7安装GPU版本tensorflow(CUDA 9.0+cudnn7.4.1)GPU(NVIDIA Quarter K2000)_第6张图片
出现这种情况表示安装完成,否则安装失败,自行排查问题
如果出现DLL load Error的问题,注意是版本的问题或者是CUDA和cuDNN版本号不对应
然后正式测试该环境是否使用的是GPU训练的
在此虚拟环境下面打开jupyter lab或者jupyter notebook
输入代码

import tensorflow as tf
import keras
# Creates a graph.
a = tf.constant([1.0, 2.0, 3.0, 4.0, 5.0, 6.0], shape=[2, 3], name='a')
b = tf.constant([1.0, 2.0, 3.0, 4.0, 5.0, 6.0], shape=[3, 2], name='b')
c = tf.matmul(a, b)
# Creates a session with log_device_placement set to True.
sess = tf.Session(config=tf.ConfigProto(log_device_placement=True))
# Runs the op.
print(sess.run(c))

运行后
如果控制台打印的日志出现
Windows 7安装GPU版本tensorflow(CUDA 9.0+cudnn7.4.1)GPU(NVIDIA Quarter K2000)_第7张图片
这种信息表示安装成功。

你可能感兴趣的:(python深度学习之路)