win10上安装GPU_tensorflow

     新系统是win10,按照网上的教程安装GPU版本的tensorflow,1.  先安装Anaconda 3.6  https://www.anaconda.com/download/#windows ;2打开 Anaconda prompt, 安装GPU的Tensor Flow,pip install --upgrade --ignore-installed tensorflow-gpu ;
3 安装CUDA, 我安装的是win10的CUDA 8.0,然后下载cudnn库文件,https://developer.nvidia.com/rdp/cudnn-download,有好几个可用于CUDA 8.0的,例如cuDNN vx.0 Library for Windows 10,最开始下载的最新版本的cudnn库文件,解压后将bin文件夹中的cudnn64_x.dll拷贝到能通过环境变量path找到的路径下。这里是拷贝到了NVIDIA GPU Computing Toolkit\CUDA\v8.0\bin目录下。但是在Anaconda 3.6 自带的spyder(python3.6)里,菜单file-new,复制粘贴以下运行测试程序:
import tensorflow as tf

hello = tf.constant('Hello world, TensorFlow!')

sess = tf.Session()

print(sess.run(hello))

后报错ModuleNotFoundError: No module named '_pywrap_tensorflow_internal',于是又下载了cuDNN v5.1 Library for Windows 10,把bin里面的dll文件拷贝后还是报这个错误,最后下载了cuDNN v6.0 Library for Windows 10,cudnn64_x6.dll拷贝到NVIDIA GPU Computing Toolkit\CUDA\v8.0\bin目录下(这里的本机目录为C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v8.0\bin),运行成功。
  • 2
    点赞
  • 3
    收藏
    觉得还不错? 一键收藏
  • 打赏
    打赏
  • 2
    评论
评论 2
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包

打赏作者

xqlily

鼓励原创

¥1 ¥2 ¥4 ¥6 ¥10 ¥20
扫码支付:¥1
获取中
扫码支付

您的余额不足,请更换扫码支付或充值

打赏作者

实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

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

余额充值