一,本机的配置:
二,文件准备。
VS2015,官网社区版(2012,2013,2017也支持),注意CUDA8.0不支持VS2017
CUDA9.0,官网下载地址
cuDNN V7.05,官网下载,需要填调查表,进去后可看历史版本
Anaconda3-5.0.1,清华镜像下载
Tensorflow,Github下载
下载页面如上,具体位置参见后文。
注意:安装CUDA9.0前请确认VS2015已安装!
三,安装,
1. VS2017
只需安装C++组件,见下图(引用自参考1),一路默认就行。
2. CUDA9.0
(1) 选择对应的版本进行下载,可选择在线安装版和离线安装版,请随意。 最好使用管理员权限进行安装,一路默认就行,安装成功界面如下图:
同时,环境变量中会多出两个路径,即:CUDA_PATH和CUDA_PATH_V8.0
如下图所示:
之后,将解压后的Cudnn文件夹中的bin、include和lib文件夹复制到下图所示的文件夹中:还要在PATH中,添加
至此,CUDA和CuDNN的配置已完成
(2) 验证CUDA安装是否成功:
打开cmd,输入:nvcc -V
出现如下类似信息:
(2)Python中TensorFlow-gpu的安装(如上图)
最好还是使用Anconda(包含Python3.6),去安装TensorFlow。
安装TensorFlow的gpu版本,只需要体格命令: