职业IT人-IT人生活圈

 找回密码
 成为会员
搜索
查看: 1417|回复: 1

Windows10下安装TensorFlow GPU版

[复制链接]
无处不在 发表于 2017-7-1 17:52 | 显示全部楼层 |阅读模式
为了好好学习,特意买了nvidia显卡的笔记本。现将Windows10下安装TensorFlow GPU版发布如下:

  • 下载 CUDA 8.01cuDNN
  • 安装CUDA
  • 安装解压cuDNN到一个你喜欢的位置,复制cuda文件夹中的文件到CUDA\v8.0目录中覆盖,同时要把bin目录中的cudnn64_6.dll复制一份改成cudnn64_5.dll,不然python不支持导入的时候会报错(类似ImportError: DLL load failed: The specified module could not be found.这种错误)。

  • 安装:pyhton,python for windows下载地址:https://www.python.org/downloads/windows/

  • 安装 tf


  pip3 install tensorflow-gpu


  过程根据网速时间较长,如出现连接超时啥的就会出错了,可以先看看论坛;)


  6.验证TensorFlow 是否安装成功。
     a.打开windows命令窗口,输入python
    b.在python中输入以下测试程序代码
        import tensorflow as tf
        hello=tf.constant('Hello, TensorFlow!')
        sess=tf.Session()
        print(sess.run(hello))

    如果能输出 Hello,TensorFlow!就代表成功。
    也可以输入tf.__version__查看tensorflow的版本
    tf.__path__查看tensorflow的安装路径。

    没有安装过vc2015的同学到微软官网下载安装可以解决python下不能导入tensorflow的问题。
    Microsoft Visual C++ 2015下载地址:https://www.microsoft.com/en-us/download/details.aspx?id=53587


您需要登录后才可以回帖 登录 | 成为会员

本版积分规则

QQ|手机版|小黑屋|网站帮助|职业IT人-IT人生活圈 ( 粤ICP备12053935号-1 )|网站地图
本站文章版权归原发布者及原出处所有。内容为作者个人观点,并不代表本站赞同其观点和对其真实性负责,本站只提供参考并不构成任何投资及应用建议。本站是信息平台,网站上部分文章为转载,并不用于任何商业目的,我们已经尽可能的对作者和来源进行了通告,但是能力有限或疏忽造成漏登,请及时联系我们,我们将根据著作权人的要求立即更正或者删除有关内容。

GMT+8, 2024-4-19 01:55 , Processed in 0.135735 second(s), 20 queries , Gzip On.

Powered by Discuz! X3.4

Copyright © 2001-2021, Tencent Cloud.

快速回复 返回顶部 返回列表