(2018最新)win10+CUDA9.0+cudnn7.1.4+TensorFlow1.8.0安装

a 夏天 提交于 2019-12-03 08:23:13

【时间】2018.11.30

【题目】win10+CUDA9.0+cudnn7.1.4+TensorFlow1.8.0安装

概述

本文主要讲述了win10+CUDA9.0+cudnn7.1+TensorFlow1.8.0安装。下面是官方教程,是英文版的。

CUDA:https://docs.nvidia.com/cuda/cuda-installation-guide-microsoft-windows/index.html

cuDNN:https://docs.nvidia.com/deeplearning/sdk/cudnn-install/index.html#installwindows

CUDA9.0+cudnn7.1.4 下载链接 :链接:https://pan.baidu.com/s/1C56ubJn1PTFYGTlYgRsemA   提取码:10fb

一、安装的各个版本

    tensorflow 1.8.0

 

下载地址:

【注意】本人在下载 exe(local)版本时,总是在最后的时候下载失败,所以最后选择了下载exe(network)版本,只有十几M。

 

【注意】在此网站中下载需要先注册账号,用QQ邮箱即可注册。

 

  •  本文选择安装tensorflow 1.8.0版本,安装命令为:pip install tensorflow_gpu==1.8.0,注意截止至本文时间,tensorflow_gpu已经更新到了1.11.0,如果直接使用pip install tensorflow_gpu安装,会安装最新版本,可能会不兼容。与各CUDA对应的
  • tensorflow版本如下:

PS:图片来源为:https://www.tensorflow.org/install/source,Linux与Mac的对应版本也可在此网址中查看。

 

二、CUDA9.0安装与环境配置

1、安装

下载好安装包后,打开直接点击‘next’下一步进行安装,安装(提取)路径默认为C盘就好。因为这只是文件的提取路径而已,安装完成后会自动删除,最后都是安装在C盘下的 C:\Program Files\NVIDIA Corporation和C:\Program Files\NVIDIA GPU Computing  Toolkit 文件夹中。

 

 

2.添加环境变量(参考链接:https://blog.csdn.net/m0_37870649/article/details/79327178

安装结束后,我们在计算机上点右键,打开属性->高级系统设置->环境变量,可以看到系统中多了CUDA_PATH和CUDA_PATH_V9_0两个环境变量。

我们还需要在环境变量中添加如下5个变量:

CUDA_SDK_PATH = C:\ProgramData\NVIDIA Corporation\CUDA Samples\v9.0

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

设置完成之后,我们可以打开cmd来查看刚才设置的环境变量: 

 

下一步是监测cuda安装成功与否:

在cuda安装文件夹C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v9.0\extras\demo_suite中有deviceQuery.exe 和 bandwidthTest.exe两个程序。

首先启动cmd DOS命令窗口,默认进来的是c:\users\Admistrator>路径,输入 cd .. 两次,来到c:目录下输入dir 找到安装的cuda文件夹。 分别输入deviceQuery.exe 和 bandwidthTest.exe,运行结果如图所示。Rsult=PASS则说明通过,反之,Rsult=Fail 则需要重新安装。 

如果以上都没问题,则说明CUDA9.0安装成功。

【注意】这时可以使用命令nvidia-smi查看GPU的使用情况,不过需要cd 到C:\Program Files\NVIDIA Corporation\NVSMI ,或者将C:\Program Files\NVIDIA Corporation\NVSMI添加到环境变量Path中,结果如下:

三、安装cudnn7.1

1、解压压缩包,把压缩包中bin,include,lib中的文件分别拷贝到C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v9.0目录下对应目录中。

即分别拷贝到的文件夹如下:

C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v9.0\bin

C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v9.0\include

C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v9.0\lib\x64

2、把C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v9.0\extras\CUPTI\libx64\cupti64_80.dll 拷贝到C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v9.0\bin (本人安装时没有进行这一步)

四、设置电脑使用GPU运行程序(一般不用设置)

一般安装好tensorflow-gpu后,当运行机器学习程序时,系统会自动调用GPU运行代码。但是有时系统不会自行设置或者但你电脑拥有多块GPU时,可以使用一下方法设置使用的GPU。参考链接:http://www.ghost580.net/article/9443.html

在图像设置中进行设置,你可以搜索‘图像设置’即可找到。

第一个下拉菜单提供您选择经典应用程序或通用应用程序。 如果您选择经典应用程序,则需要手动浏览并选择该应用程序的EXE文件。 如果您选择UWP应用程序,则会在您的PC上提供已安装应用程序的列表。

 

 

 

填充完列表后,选择要自定义图形性能的应用程序,然后选择选项。 下一个窗口会根据性能列出显卡。 您应该使用它们的名称来节省GPU和高性能GPU。

 

选择,系统默认,省电和高性能。 然后保存它。其中省电就是使用原本电脑的GPU,而高性能则是使用附加的GPU如1080Ti等。

PS1:暂时没有找到在WIN10中指定程序运行使用特定的GPU的方法,在Linux中使用的CUDA_VISIBLE_DEVICES=0  python code.py的方法并不适用,只能在代码中设置,例如:

#指定第一块GPU可用

os.environ["CUDA_VISIBLE_DEVICES"] = "0" 

config = tf.ConfigProto()

#不全部占满显存, 按需分配

config.gpu_options.allow_growth=True

sess = tf.Session(config=config)

KTF.set_session(sess)

PS2:除了使用nvidia-smi查看GPU使用情况外,还可以在任务管理器----性能中查看:

 

 

 

易学教程内所有资源均来自网络或用户发布的内容,如有违反法律规定的内容欢迎反馈
该文章没有解决你所遇到的问题?点击提问,说说你的问题,让更多的人一起探讨吧!