win10搭建tensorflow-gpu环境

昨天辛苦的配了GPU环境,记录一下防止之后还须要用到。ubuntu

我配GPU的目的是用tensorflow的gpu来加速windows

不用ubuntu是由于一来不习惯,二来我不会配ubuntu的扩展显示器,就更不习惯了,习惯了两个屏以后,一个屏看的着实有点难受框架

1、本身的环境ui

操做系统:win10   GPU:GTX1050+CUDA9.0+cuDNN    IDE:Pycharm    框架:tensorflow-gpu    解释器:Python3.6(强烈建议anaconda3,又方便又强大,下载太慢能够找镜像)操作系统

 

2、安装顺序3d

一、首先安装CUDA9.0。下载地址。blog

选择9.0版本是由于网上说,tensorflow的windows-gpu还不支持CUDA的新版本。再加上同窗安装的也是9.0,用的没有问题,因此就选择了这个版本。教程

在下载的时候选择适合本身的版本。     ip

我选择的如上图,刚开始第四行选择的是右边的那个,可是下载下来安装不了,百度查是电脑缺乏什么压缩软件,因此建议选左边的那个。v8

而后只须要选择下图的base installer便可。

 

打开下载好的CUDA9.0开始安装,刚开始的安装程序存放位置,本身随意就好。

下一步会检测系统兼容性,有些显卡是不支持GPU的,本身须要先查清楚。

下一步接受协议

下一步选择安装模式,我选择的是自定义,程序默认的精简模式应该能够理解为安装全部东西,其中包括了我暂时不用的VS以及显卡驱动,因此我选择的是自定义模式。这也是查了其余的博文所作出的选择。

下一步的选择为

 

下一步会让选择安装路径,我最初是本身设置了路径,由于不想往C盘放东西,但是安装完以后,在我设置的路径里没找到安装的CUDA,他仍是安装在了C盘,而且没有安装彻底。因此我安装了第二次,我并无卸载第一次装的,由于没找到卸载的方法。直接开始了第二次安装,仍是选择本身设置的路径,仍是没安装成功。很难受,因此第三次安装直接选择他的默认路径,一点多个G还能接受,放在C盘也问题不大。此次安装成功了。

因此我的建议,若是C盘能接受一个多G,不如直接安装在默认路径,防止老出问题。

安装完成以后,下一步须要配置环境变量。

若是安装成功,系统变量会多出两个为

还须要本身添加几个系统变量

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

添加好以后应该像下面这样。

下一步在系统变量PATH里添加东西。找到系统变量的PATH双击。

添加下列东西。

%CUDA_LIB_PATH%

%CUDA_BIN_PATH%

%CUDA_SDK_LIB_PATH%

%CUDA_SDK_BIN_PATH%

C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v8.0\lib\x64   这些均为默认路径,有须要的话自行修改

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

C:\ProgramData\NVIDIA Corporation\CUDA Samples\v8.0\common\lib\x64

C:\ProgramData\NVIDIA Corporation\CUDA Samples\v8.0\bin\win64

添加完以后CUDA就算安装完成了。咱们能够检验是否安装成功。

打开CMD,cd到C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v9.0\extras\demo_suite    (这是默认路径)

分别执行命令:bandwidthTest.exe

                         deviceQuery.exe

若分别返回

 

则表明CUDA9.0安装成功。

二、安装cudnn。

官网上下载搭配CUDA9.0的cudnn ,必定要注意搭配9.0。会须要注册英伟达的帐号。

下载好以后,解压,分别将.h .lib 和.dll 文件分别拷贝到cuda的include, lib/x64, bin 文件夹下。其实也就是对应的文件夹。

到这里就安装好了CUDA9.0及其对应的cudnn。

三、安装tensorflow-gpu

 以前安装的是cpu版本,因此先删除。 pip uninstall tensorflow

删除完以后,习惯性的觉得 pip install tensorflow-gpu就能够了,结果报了一堆错。我没有尝试去下载whl文件安装,你能够试试。我看的那篇教程说whl也有错

反正我后来是删了anaconda而后重装了一下,对了,安装anaconda的时候不要手贱关掉跳出来的东西,我第一次重装的时候就没装全,由于本身没注意瞎点了跳出来的终端框

而后conda install tensorflow-gpu就行了。因此我怀疑pip命令无论用,conda可能更管用点,建议删除完cpu版本时候先试下conda的命令,若是也报错,再考虑其余方法。

 这是安装好以后的终端。泪流满面。

经过下图中的代码能够检验tensorflow-gpu是否安装成功。返回差很少的界面表明成功

 

 

 

以上是配置tensorflow-gpu的经历,但愿能帮到你。

小白一枚,有错误欢迎指出。