这篇文章主要讲TensorFlow在原生Windows下的安装教程。
安装环境:TensorFlow0.12+cuda8.0+cudnn v5.1+window7+python3.5
(1) 先安装Python3.5
从官网下载Python3.5,https://www.python.org/downloads/windows/,双击安装即可。接着将安装路径添加进环境变量中。具体步骤:计算机右键属性—高级系统设置—环境变量—系统变量—找到Path,把Python安装路径添加进去。
(2)安装Cuda和Cudnn(CPU版本的不需要安装,可直接跳过第二步)
首先要检查电脑的显卡型号是否支持GPU加速。
若出现上图结果,即说明你安装cuda成功。
(3)安装tensorflow
在安装TensorFlow之前,先修改镜像
国外镜像使用pip安装速度很慢,为了使用国内镜像加速pip安装,需要如下修改:
Windows 7在“C:/Users/用户名/AppData/Local/pip”文件夹下,新建文本文件,添加内容:
1. [global]
2. index-url = http://mirrors.aliyun.com/pypi/simple/
3. [install]
4. trusted-host=mirrors.aliyun.com
ps:这里使用的是阿里云的国内镜像。
接着,在开始菜单打开windows PowerShell,输入py -3 –m pip install --upgrade https://storage.googleapis.com/tensorflow/windows/gpu/tensorflow_gpu-0.12.0rc0-cp35-cp35m-win_amd64.whl
Ps:因为本人的电脑同时装了Python2.7和Python3.5,所以必须指明是使用哪一个版本的pip模块,若读者电脑只装了Python3.5,那么前面的py -3 –m语句可以忽略。
安装完成后打开Python 3.5 64bit 进行测试。
到“所有程序”下找到"Python 3.5 64bit",出现命令窗口,输入测试代码:
1. >>>import tensorflow as tf
2. >>>sess = tf.session()
3. >>>x = tf.constant(5)
4. >>>y = tf.constant(35)
5. >>>print(sess.run(x * y))
6. 175
测试成功,TensorFlow安装完成。
最后,检验一下cudnn是否配置成功:
Windows PowerShell 下输入python(因为我的电脑同时装了Python2.7和3.5,所以必须用py -3指定用哪个版本,若读者只装了Python3.5,那么直接输入python即可),再输入import tensorflow,若出现下图所示,不提示没有安装cudnn,则说明cudnn配置成功。
以上,便是本人在安装TensorFlow时的具体流程,如有差错,还望不吝赐教!
参考:
原生Windows安装TensorFlow 0.12方法
windows 10 64bit下安装Tensorflow+Keras+VS2015+CUDA8.0 GPU加速
新闻热点
疑难解答