职业IT人论坛

 找回密码
 注册

扫一扫,访问微社区

搜索
查看: 824|回复: 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人 ( 粤ICP备12053935号 )

GMT+8, 2018-8-20 17:16 , Processed in 0.070384 second(s), 18 queries , Gzip On.

Powered by Discuz! X3.2

© 2001-2018 Comsenz Inc.

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