win10上安装GPU_tensorflow

来源:互联网 发布:java访问次数 编辑:程序博客网 时间:2024/05/23 07:23
     新系统是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 tfhello = 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),运行成功。