深度学习框架之tensorflow-gpu配置

深度学习环境配置

软硬件配置:
GTX 1070ti
Win10 64位专业版
Anaconda3-5.2.0-Windows-x86_64

深度学习工具:
CUDA 9.0
cuDNN v7.1 注:版本必须匹配CUDA9.0
TensorFlow 1.8.0(GPU版)

第一步:安装vs2015
下载“ [Visual Studio Community 2015][with Update 3][官方简体]”。
安装类型:选择自定义,点击下一步。
这里写图片描述

在编程语言(Programming Languages)中只需安装C++组件,点击下一步,安装位置可以自选,其他全部按照默认。
这里写图片描述

第二步:安装CUDA9.0
Google:CUDA Toolkit 9.0 Downloads | NVIDIA Developer 选择对应自己电脑环境的版本下载。下载之后安装即可。
这里写图片描述

第三步:安装cuDNN v7.1

cuDNN的全称为NVIDIA CUDA® Deep Neural Network library,是NVIDIA专门针对深度神经网络(Deep Neural Networks)中的基础操作而设计基于GPU的加速库。cuDNN为深度神经网络中的标准流程提供了高度优化的实现方式,例如convolution、pooling、normalization以及activation layers的前向以及后向过程。

cuDNN只是NVIDIA深度神经网络软件开发包中的其中一种加速库。想了解NVIDIA深度神经网络加速库中的其他包请戳链接https://developer.nvidia.com/deep-learning-software

安装重点:点击网址进入https://developer.nvidia.com/cudnn,下载cudnn要先注册这个网站,并且cudnn的版本必须和CUDA的版本匹配。
这里写图片描述

第四步:安装Anaconda3-5.2.0
进入网址https://www.anaconda.com/download/
这里写图片描述
下载Python 3.6版本,点击downloads,下载之后安装即可,注意安装快结束的时候两个选项的对勾选上,其中一个是设置环境变量。

第五步:安装tensorflow(gpu)
Anaconda的Anaconda prompt中输入conda install tensorflow-gpu命令即可安装gpu版本的tensorflow。

检验CUDA是否正确安装
方法一:
在运行中输入cmd,点击确定,cd至文件夹:C:\Program Files\NVIDIA Corporation\NVSMI,点击确认。输入nvidia-smi
这里写图片描述

方法二:
在运行中输入cmd,点击确定,输入nvcc –V,点击确定。
这里写图片描述

检验tensorflow是否能够使用
打开Anaconda navigator,在Home中设置Applications on tensorflow,并且安装spyder。安装完成后会在Anaconda中出现spyder(tensorflow)
这里写图片描述
这里写图片描述
打开spyder,输入下列代码测试:
import tensorflow as tf
import numpy as np
import matplotlib.pyplot as plt

train_x=np.random.random(100)
train_y=2*train_x-1+np.random.randn(*train_x.shape)*0.3 #y=2x
plt.plot(train_x,train_y,’ro’,label=’Original data’)
plt.legend()
plt.show()

这里写图片描述
能够执行,则说明tensorflow安装成功

猜你喜欢

转载自blog.csdn.net/weixin_42920648/article/details/81531280
今日推荐