Win7 + tensorflow-gpu1.8.0 + cuda9.0 + cuDNN 7.0 + Python3.5.2 + vs2013安装教程

386 阅读3分钟

#Win7 + tensorflow-gpu1.8.0 + cuda9.0 + cuDNN 7.0 + Python3.5.2 + vs2013安装教程

如需转载请标明出处
QQ技术交流群:129518033

目录

### 文章目录* * #Win7 + tensorflow-gpu1.8.0 + cuda9.0 + cuDNN 7.0 + Python3.5.2 + vs2013安装教程

tensorflow:tensorflow-gpu1.8.0
cuda:cuda_9.0.176_windows
cuDNN:cuDNN v7.1.4 (May 16, 2018), for CUDA 9.0
python:3.5.2 64bit
编译器:VS2013 12.0.21005.1 REL
显卡:NVIDIA GeForce GTX 950
系统环境:Windows 7 64bit


##前言
tensorflow与python的结合简直是机器学习的完美搭配,本文就介绍如何搭建tensorflow的gpu开发环境。

##1.VS2013安装和Python 3.5.2安装
略过

##2.cuda9.0指南知识
CUDA 9.0的安装指南:
docs.nvidia.com/cuda/archiv…

###2.1 CUDA 9.0操作系统支持

操作系统Native x86_64Cross (x86_32 on x86_64)
Windows 10YESYES
Windows 8.1YESYES
Windows 7YESYES
Windows Server 2016YESNO
Windows Server 2012 R2 (*)YESNO

###2.2 CUDA 9.0 window下编译器支持

CompilerIDENative x86_64Cross (x86_32 on x86_64)
Visual C++ 15.0Visual Studio 2017YESNO
Visual C++ 14.0Visual Studio 2015YESNO
Visual C++ 14.0Visual Studio Community 2015YESNO
Visual C++ 12.0Visual Studio 2013YESYES
Visual C++ 11.0Visual Studio 2012YESYES
Visual C++ 10.0 DEPRECATEDVisual Studio 2010YESYES

##3.CUDA 9.0安装
###3.1下载CUDA9.0
下载地址:
developer.nvidia.com/cuda-90-dow…

这里写图片描述

这里写图片描述

###3.2安装主程序
点击cuda_9.0.176_windows.exe进行安装
这里写图片描述

如图,1表示主程序,剩余3个为补丁程序,按照顺序安装即可。

这里写图片描述

点击同意并继续

这里写图片描述
点击自定义(高级)

这里写图片描述
全部勾选,默认即为全部勾选

这里写图片描述

点击下一步

这里写图片描述

这里写图片描述

程序开始安装中…

这里写图片描述

安装完成

###3.3安装补丁程序
补丁程序按照顺序安装即可
这里写图片描述

这里写图片描述

##4.cuDNN安装

版本:
cuDNN v7.1.4 (May 16, 2018), for CUDA 9.0

下载地址:
developer.nvidia.com/rdp/cudnn-d…

这里写图片描述

###4.1 解压文件
下载完成后解压cudnn-9.0-windows7-x64-v7.1.zip

###4.2 覆盖文件
将文件夹里的内容拷贝到CUDA的安装目录并覆盖相应的文件夹,CUDA拷贝目录:C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v9.0

这里写图片描述

这里写图片描述

##5.测试sample

测试程序目录:
C:\ProgramData\NVIDIA Corporation\CUDA Samples\v9.0

推荐sample:3_Imaging\histogram

这里写图片描述

其他sample运行失败也无所谓,因为部分Sample依赖了第三方库,所以很多Sample无法正常编译。

##6.安装tensorflow-gpu1.8.0

这里下载的是离线的whl文件,版本为gpu1.8.0

###6.1 下载
下载地址:tensorflow-gpu1.8.0

###6.2 离线安装

pip install c:\tensorflow_gpu-1.8.0-cp35-cp35m-win_amd64.whl

这里写图片描述

###6.3 安装完成

这里写图片描述

##7.tensorflow的GPU验证

验证代码

import tensorflow as tf

a = tf.test.is_built_with_cuda()  # 判断CUDA是否可以用

b = tf.test.is_gpu_available(
    cuda_only=False,
    min_cuda_compute_capability=None
)                                  # 判断GPU是否可以用

print(a)
print(b)

运行结果:

2018-07-17 22:40:19.814200: I T:\src\github\tensorflow\tensorflow\core\platform\cpu_feature_guard.cc:140] Your CPU supports instructions that this TensorFlow binary was not compiled to use: AVX2
2018-07-17 22:40:20.025200: I T:\src\github\tensorflow\tensorflow\core\common_runtime\gpu\gpu_device.cc:1356] Found device 0 with properties: 
name: GeForce GTX 950 major: 5 minor: 2 memoryClockRate(GHz): 1.2155
pciBusID: 0000:01:00.0
totalMemory: 2.00GiB freeMemory: 1.89GiB
2018-07-17 22:40:20.025200: I T:\src\github\tensorflow\tensorflow\core\common_runtime\gpu\gpu_device.cc:1435] Adding visible gpu devices: 0
2018-07-17 22:40:20.347200: I T:\src\github\tensorflow\tensorflow\core\common_runtime\gpu\gpu_device.cc:923] Device interconnect StreamExecutor with strength 1 edge matrix:
2018-07-17 22:40:20.348200: I T:\src\github\tensorflow\tensorflow\core\common_runtime\gpu\gpu_device.cc:929]      0 
2018-07-17 22:40:20.348200: I T:\src\github\tensorflow\tensorflow\core\common_runtime\gpu\gpu_device.cc:942] 0:   N 
2018-07-17 22:40:20.348200: I T:\src\github\tensorflow\tensorflow\core\common_runtime\gpu\gpu_device.cc:1053] Created TensorFlow device (/device:GPU:0 with 1668 MB memory) -> physical GPU (device: 0, name: GeForce GTX 950, pci bus id: 0000:01:00.0, compute capability: 5.2)
True
True

这里写图片描述


Refrence:
docs.nvidia.com/cuda/archiv…
docs.nvidia.com/deeplearnin…
blog.csdn.net/weixin_4235…

如需转载请标明出处
QQ技术交流群:129518033