筒子们先注意,不要下载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