参考网址:https://tensorflow.google.cn/install/gpu
Tensorflow-gpu不同版本要求与CUDA及CUDNN版本对应关系
1.升级显卡驱动
右击 计算机 ,选择 管理
设备管理器→显示适配器→更新驱动程序软件
等待更新完成
2.查看本机支持CUDA的版本
右击桌面→打开NVIDIA控制面板 或者 开始→控制面板→NVIDIA控制面板
帮助→系统信息→组件 查看CUDA支持版本
3. 下载CUDA9及对应Cudnn7
安装包地址:https://pan.baidu.com/s/1ONri0GoER2R6hu0sIXjyZw
提取码:zepr
如果上述地址中没有你需要的,可在官网下载
CUDA下载网址:https://developer.nvidia.com/cuda-toolkit-archive
Cudnn网址:https://developer.nvidia.com/rdp/cudnn-archive
Cudnn下载需要注册,注册后进行下载
4. 卸载CUDA8(之前没装CUDA可忽略此步骤)
留下NVIDIA重要的2个软件:NVIDIA的图形驱动软件,NVIDIA Physx系统软件,其他的都删了。C盘C:\Program Files\NVIDIA GPU Computing Toolkit这个文件也可以删除,然后清理垃圾。
开始→控制面板→程序→程序卸载
将版本为8.0的全部卸载,无先后顺序(上图为CUDA9.1示意图)
5.安装CUDA9
下面是安装过程(按照默认安装):
接下来就是同意继续下一步,等待CUDA安装的结束。
6.设置环境变量
安装结束后,我们在计算机上点右键,打开属性->高级系统设置->环境变量,可以看到系统中多了CUDA_PATH和CUDA_PATH_V9_1两个环境变量。
我们还需要在环境变量中添加如下几个变量:
CUDA_SDK_PATH = C:\ProgramData\NVIDIA Corporation\CUDA Samples\v9.1
CUDA_LIB_PATH = %CUDA_PATH%\lib\x64
CUDA_BIN_PATH = %CUDA_PATH%\bin
CUDA_SDK_BIN_PATH = %CUDA_SDK_PATH%\bin\win64
CUDA_SDK_LIB_PATH = %CUDA_SDK_PATH%\common\lib\x64
验证:
7.检查cuda安装成功与否
cmd目录下 运行 nvcc -V
出现上述所示表示安装成功
其它验证方法:
在cuda安装文件夹中有deviceQuery.exe 和 bandwidthTest.exe两个程序。首先启动cmd DOS命令窗口,默认进来的是c:\users\Admistrator>路径,输入 cd … 两次,来到c:目录下输入dir 找到安装的cuda文件夹。
分别输入deviceQuery.exe 和 bandwidthTest.exe,运行结果如图所示。Rsult=PASS则说明通过,反之,Rsult=Fail 则需要重新安装。
8.配置CUDNN
解压得到
a) 把cuda\bin\cudnn64_7.dll复制到 C:\Program Files
\NVIDIA GPU Computing Toolkit\CUDA\v9.0\bin 目录下.
b) 把\cuda\ include\cudnn.h复制到 C:\Program Files
\NVIDIA GPU Computing Toolkit\CUDA\v9.0\include 目录下.
c) 把\cuda\lib\x64\cudnn.lib复制到C:\Program Files
\NVIDIA GPU Computing Toolkit\CUDA\v9.0\lib\x64 目录下.
CUDA安装已完成
9.安装tensorflow 1.11
GPU
cmd 输入 Pip install tensorflow-gpu=1.11 等待完成
若出现错误,可直接安装wheel包
地址为:https://pan.baidu.com/s/1ONri0GoER2R6hu0sIXjyZw
提取码:zepr
将whl文件复制到本地,cmd转到whl文件所在文件夹下,输入pip install tensorflow_gpu-1.11.0-cp36-cp36m-win_amd64.whl,等待安装完成。
CPU
cmd 输入 Pip install tensorflow=1.11 等待完成
若出现错误,可直接安装wheel包
地址为:https://pan.baidu.com/s/1ONri0GoER2R6hu0sIXjyZw
提取码:zepr
将whl文件复制到本地,cmd转到whl文件所在文件夹下,输入pip install tensorflow-1.11.0-cp36-cp36m-win_amd64.whl,等待安装完成。
验证tensorflow环境是否安装成功
检测目前安装了哪些环境:conda info --envs
activate tensorflow-gpu 进入gpu环境
deactivate 退出环境
activate tensorflow 进入cpu环境
deactivate 退出环境