Tensorflow+cuda+cudnn+window+Python之window下安装TensorFlow

这篇文章主要讲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加速。

https://developer.nvidia.com/cuda-gpus点击CUDA-EnabledGeForce Products。好像是computer capability大于3.0的才支持。我的显卡版本是GT720(低端显卡,凑合着用),支持GPU加速。

确定电脑显卡支持GPU加速后,开始下载Cuda: https://developer.nvidia.com/cuda-downloads,(这里有个坑,如果浏览器下载一直下不了的话,建议用百度云离线下载)

下载完直接双击安装即可。

接下来下载Cudnn: https://developer.nvidia.com/cudnn

Cudnn解压后将bin,include,lib三个文件夹里面的内容覆盖至Cuda安装目录下,默认路径为C:\Program Files\NVIDIA GPUComputing Toolkit\CUDA\v8.0(记住不是替换,是把Cudnn文件里的.dll文件添加到Cuda里面)

接下来,检验Cuda是否安装成功:

进入cuda路径,默认C:\ProgramData\NVIDIACorporation\CUDA Samples\v8.0\bin\win64\Release执行里面的deviceQuery.exe文件,如下图所示:


                                                                   若出现上图结果,即说明你安装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加速


   

 

 

 

  • 0
    点赞
  • 8
    收藏
    觉得还不错? 一键收藏
  • 4
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论 4
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值