win10+TensorFlow gpu1.7+cuda9.0+cudnn7.0+配置教程

筒子们先注意,不要下载CUDA9.1!!!不要下载9.1版本 因为貌似TensorFlow的兼容性不是很好,我也是刚打算学习TensorFlow框架,一开始下载了9.1折腾了好久也没结果,后面降级才测试成功。比较稳定的是CUDA9.0+cunnd7.0+TensorFlow1.7版本。

网上很多同学都写了cpu版本的安装教程,GPU的教程多少都有些不完整

下面进入正题:

第一步:做好相关环境工具的下载:分别是:

1.cuda_9.0.176_win10 

下载地址:链接:https://pan.baidu.com/s/1rm7imUVnXw2x68nbBbaZEA 密码:wj1i

2.cudnn-9.0-windows10-x64-v7 (cudnn7.0版本)

下载地址:链接:https://pan.baidu.com/s/1V1CQpNwocajOyPTUlSFDyA 密码:s5ig

3.tensorflow_gpu-1.7.0-cp35-cp35m-win_amd64.whl 

下载地址:链接:https://pan.baidu.com/s/1f4IlIBnMirY2OPhKvhMtqQ 密码:84b1

其中第三个也可以用pip在线下载,只不过网速比较慢就直接放上来方便同学下载

另外需要下载安装anacanda3,具体安装anacanda的教程在下:

https://blog.csdn.net/u012318074/article/details/77075209

第二步开始环境的搭建:

1.分别安装anacanda3,cuda_9.0两个软件。

2.将cudnn-9.0-windows10-x64-v7解压后拷贝到下图的文件夹里面

3.进行环境搭建

3.1 国内的同学可以先换成清华的源,这样下载起来快一些。

在anacanda3的应用中找到Anaconda Prompt的应用(其实就是终端)打开后输入:

conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/
conda config --set show_channel_urls yes

3.2 安装tensorFlowgpu版本,同时配置python3.5的环境

conda create -n tensorflow-gpu python=3.5

3.3 激活tensorflow

activate tensorflow-gpu

3.4 升级到最新版本的pip10.0.0,之前我没有升级,然后在安装whl文件过程中不断的提示错误。

python -m pip install --upgrade pip

3.5安装tensorflow_gpu-1.7.0-cp35-cp35m-win_amd64.whl文件 

安装这个文件可以在线安装,或者本地安装百度云分享的文件,因为可能在线安装会掉线或者速度很慢。

在线安装直接输入命令:

pip install --upgrade --ignore-installed tensorflow-gpu 

本地安装就是把文件放在系统盘里,然后通过终端进入相对应的目录输入命令:

pip install tensorflow_gpu-1.7.0-cp35-cp35m-win_amd64.whl

等待安装完成后就完成环境的配置了,下一步可以进入测试阶段。

第三步:测试环境

这里要说明一下 上述以及下述的过程都是在激活tensorflow-gpu的情况下进行的。

python //进入python命令模式
import tensorflow as tf
hello=tf.constant("Hello,tensorflow!")
s=tf.Session()
print (s.run(hello))
顺利打印出b"Hello,tensorflow!"说明成功,
exit()//退出python命令模式

之后就可以通过其他IDE来使用TensorFlow啦,比如说pycharm。

我也是刚开始接触深度学习的小学生,如果还有不清楚的同学可以留言讨论。感谢本文借鉴的文章作者,部分引用如下:

https://blog.csdn.net/u013055678/article/details/59107932

https://blog.csdn.net/zhichitiany2014/article/details/77671481

https://blog.csdn.net/u011440696/article/details/79381375

阅读更多
想对作者说点什么?

博主推荐

换一批

没有更多推荐了,返回首页