上一篇文章已经讲解了Ubuntu下CUDA+CUDNN+Tensorflow-gpu的安装,本文主要讲解windows10环境下安装Tensorflow-GPU1.4.0。
安装框架:1080Ti—win10 + cuda8.0 + cudnn6.0 + anaconda(python 3.5) + tensorflow-gpu1.4
总的来说:win的安装比Ubuntu的安装简单多了,很多差不多的步骤这里就一语带过不上图了,有关下载网址请移步(二)Win10和Ubuntu16.04双系统下Ubuntu安装Tensorflow-GPU找到对应部分即可。
0 卸载其他版本的CUDA
打开控制面板-大图标-卸载程序
你将看到很多关于Nvidia的选项,右键卸载即可,但请注意:留下"NVIDIA的图形驱动程序"、“NVIDIA Physx系统软件”,反正包含Nvidia的都可以卸了。完事后也可以把C盘的对应文件夹给删掉,不删应该也没多大事。
重启电脑!
1 下载
(二)已包含,略。
2 驱动安装
一般不需要安装,或者也可以到NVIDIA官网下载390.48版本,或者任意新版本,无所谓。
3 CUDA 8.0安装
下载安装
进入官网按照下图所示选择安装。
双击exe安装即可,可以直接选择默认安装,后面环境配置会很方便。
环境配置
右键“计算机”,选择“属性”;选择“高级系统设置”;选择”环境变量”。
Cuda8.0安装完成之后,需要配置7个环境变量。其中CUDA自动配置了2个系统变量,所以手动配置5个即可。自动配置的两个环境变量分别是:
CUDA_PATH: C:\Program Files\NVIDIA GPU ComputingToolkit\CUDA\v8.0
CUDA_PATH_V8_0:C:\Program Files\NVIDIA GPU ComputingToolkit\CUDA\v8.0
需要手动配置的5个环境变量分别是:
CUDA_BIN_PATH:%CUDA_PATH%\bin
CUDA_LIB_PATH:%CUDA_PATH%\lib\Win32
CUDA_SDK_BIN:%CUDA_SDK_PATH%\bin\Win64
CUDA_SDK_LIB:%CUDA_SDK_PATH%\common\lib\x64
下面一个路径需要显示隐藏文件夹才能看到。
CUDA_SDK_PATH:C:\ProgramData\NVIDIA Corporation\CUDASamples\v8.0
然后在用户变量的path中编辑-新建,复制以下内容后确定。
%CUDA_LIB_PATH%;%CUDA_BIN_PATH%;%CUDA_SDK_LIB_PATH%;%CUDA_SDK_BIN_PATH%;
重启电脑!
然后打开命令行,输入nvcc -V即可查看CUDA版本信息,确认是否安装成功。
4 CUDNN 6.0安装
官网下载对应版本
需要注册个邮箱,很快的。
找到Download cuDNN v6.0 (April 27, 2017), for CUDA 8.0,下载下图中第6行的那个for win10的文件,懒得打开网站截图了0.0
解压缩
复制CUDNN的相应文件到CUDA的对应目录即可(地址可参考截图地址栏):
将.h .lib 和.dll 文件分别拷贝到cuda的include, lib/x64, bin 文件夹下
至此CUDA和CUDNN安装完成。
5 Anaconda安装
(二)已包含,略。下载windows版本即可。
6 在线安装Tensorflow-GPU=1.4.0
简单,一句话:
pip install tensorflow-gpu==1.4
等他完成就行!
下面代码供测试gpu版tf是否安装成功:
python
>>>import tensorflow as tf
>>>a = tf.constant(1)
>>>b = tf.constant(2)
>>>c = a + b
>>>sess = tf.Session()
>>>sess.run(tf.global_variables_initializer())
>>>sess.run( c ) #结果输出3代表能够运行,安装成功。
哈哈哈写完啦,撒花撒花。
(写博客不易,头发都掉完了,各位看官赏个五角一块的,我好去买霸王洗发水啊 duang~_~)