一、安装cuda 9.1+VS2017
一路下一步即可,环境变量cuda会自动配好。
安装完后,在主目录搜索deviceQuery.exe,跑一跑测试下就好了(成功最下一句是:RESULT PASS)。
vs2017和cuda 9.1同时安装的,虽然cuda暂时提示找不到vs,但是不影响。
二、安装cudnn7.0
下载后解压,然后复制文件:
3. Copy the following files into the CUDA Toolkit directory.
a) Copy \cuda\bin\cudnn64_7.dll to C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v9.0\bin.
b) Copy \cuda\ include\cudnn.h to C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v9.0\include.
c) Copy \cuda\lib\x64\cudnn.lib to C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v9.0\lib\x64.
三、tensorflow安装
据说tf不支持cuda 9.1(因为官方的wheel是在cuda 9.0上编译的~)
当然有大神自己编译了一堆whl,方便不同版本使用。
https://github.com/fo40225/tensorflow-windows-wheel
找到对应cuda9.1版本,tf1.5的whl文件。
安装成功。