Win10配置tensorflow-gpu 1.14全过程

这两天装新版本tensorflow-gpu踩了好多坑,记录一下正确安装的过程,我的配置:Win10+python 3.7.2+CUDA 10.0+Cudnn7.4.2+tensorflow1.14

下载必备软件

  1. python 3.7.x 下载地址
  2. CUDA 10.0 下载地址
  3. Cudnn 7.4.2 对应CUDA 10.0版本的,需要注册登录,很烦!查看版本对应
  4. 如果电脑上没有安装过 visual studio 2015,那么需要首先到微软下载安装 Visual C++ Redistributable 下载地址

安装

  1. 首先更新显卡驱动,到NVIDIA官网上下载自己的显卡对应的最新驱动,我的显卡是960M,找到最新版本下载安装就可以替换掉旧的驱动了
  2. 安装CUDA 10.0,这一步除了磁盘空间不足,基本上都会成功,在命令行输入nvcc -V,如果显示了CUDA版本信息,说明安装没问题
  3. 解压Cudnn,解压后的cuda文件夹中包含三个子文件夹,\bin, \include, \lib,分别将cudnn64_7.dll, cudnn.h, cudnn.lib 拷贝到Nvdia安装目录下 C:\Program Files\NVDIA CPU Computing Toolkit\CUDA\v10.0 的\bin, \include, lib\X64 文件夹中。 查看系统环境变量,添加CUDA的\bin, extras\CUPTI, \libnvvp这三个路径
  4. 安装python 3.7.x,安装时勾选Add to Path
  5. 在命令行输入 pip install tensorflow -i https://pypi.doubanio.com/simple/
    在豆瓣源下载安装tensorflow,由于1.14是最新版,因此不需要选择版本

最后打开python,输入 import tensorflow as tf,如果没有报错,说明tensorflow-gpu安装成功!

问题

如果import时报错,找不到DLL这类错误,很可能是因为环境变量没有配置全,或者cuda,cudnn和tf版本不匹配,可以将报错位置的文件用dumpin.exe分析一下,找到里面的dll到底缺了哪一个,我第一次配置时,发现cusolver的DLL版本不一致,而这个的问题在显卡驱动,所以我卸载全部,更新了显卡驱动,重新安装,就解决了问题

Thank you for every coin~