win8安装tensorflow-gpu记录

番茄炒蛋

三个部分cuda、tensorflow、cudnn。
如果安装的的cpu版的tensorflow,在pycharm里包的名字就是tensorflow,有的人解决gpu版出问题的方法就是卸载gpu版然后装默认的cpu版, 也确实能跑程序。。。

下面是tensorflow和cudnn、cuda版本对应关系。参考地址:https://tensorflow.google.cn/install/source_windows

Version Python version Compiler Build tools cuDNN CUDA
tensorflow_gpu-1.11.0 3.5-3.6 MSVC 2015 update 3 Bazel 0.15.0 7 9
tensorflow_gpu-1.10.0 3.5-3.6 MSVC 2015 update 3 Cmake v3.6.3 7 9
tensorflow_gpu-1.9.0 3.5-3.6 MSVC 2015 update 3 Cmake v3.6.3 7 9
tensorflow_gpu-1.8.0 3.5-3.6 MSVC 2015 update 3 Cmake v3.6.3 7 9
tensorflow_gpu-1.7.0 3.5-3.6 MSVC 2015 update 3 Cmake v3.6.3 7 9
tensorflow_gpu-1.6.0 3.5-3.6 MSVC 2015 update 3 Cmake v3.6.3 7 9
tensorflow_gpu-1.5.0 3.5-3.6 MSVC 2015 update 3 Cmake v3.6.3 7 9
tensorflow_gpu-1.4.0 3.5-3.6 MSVC 2015 update 3 Cmake v3.6.3 6 8
tensorflow_gpu-1.3.0 3.5-3.6 MSVC 2015 update 3 Cmake v3.6.3 6 8
tensorflow_gpu-1.2.0 3.5-3.6 MSVC 2015 update 3 Cmake v3.6.3 5.1 8
tensorflow_gpu-1.1.0 3.5 MSVC 2015 update 3 Cmake v3.6.3 5.1 8
tensorflow_gpu-1.0.0 3.5 MSVC 2015 update 3 Cmake v3.6.3 5.1 8
  1. 先装cuda,到官网下载,版本选择和自己vs版本匹配即可,特别说明cuda9.2好像是没有官方编译的和tensorflow配合的版本。9.0和10.0都可以。
  2. 安装tensorflow-gpu,pycharm里 project interpreter就可以安装对应的包了,这里我选择的版本是1.10。最好再把numpy这个包也给装了,后面继续学习会用到。
  3. 安装cudnn,也是到英伟达官网下载,需要用邮箱注册账户,也是要选择和cuda匹配的版本。由于没有win8的选项,我们选择win7版的就行,别选win10.。下载完压缩包后,需要将三个文件夹(bin, lib, include)的文件复制到cuda安装路径下的对应文件夹内,我的默认路径就是C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v9.2
  4. 最后输入import tensorflow as tf 不报错就可以了。

问题

  1. 找不到cudart_.dll:cuda版本有问题。
  2. tensorflow有错误返回:最好检查cudnn的版本

你可能感兴趣的:(深度学习)