windows10 64位 Tensorflow安装--CUDA 9.1+cuDNN7.1+python3.6.4+tensorflow-GPU1.7.0安装详解

时间:2024-04-13 16:52:57

一、前言

vs2017安装请见博主文章:vs2017安装和使用教程(详细)

Anaconda安装可见博主文章:Anaconda安装


二、改动

1.我在Anaconda文章里写的Tensorflow是针对于CUDA8.0来说的,因此安装的是py3.5

2.CUDA9.0是官方Tensorflow的最新版本,那么好不容易装好的CUDA9.1就泡汤了。但是,世上无难事,只怕有心人!

3.接下来我为大家展示CUDA9.1也是可以用Tensorflow的!


三、CUDA9.1下载

1.网址:CUDA9.1下载,安装CUDA8.0的话会提示没有vs C++2015 redistributable,我们安装了vs2017,自然就要用CUDA9版本

windows10 64位 Tensorflow安装--CUDA 9.1+cuDNN7.1+python3.6.4+tensorflow-GPU1.7.0安装详解

下载之后安装过程中之后注意关闭360!!否则安装失败!!

2.查看安装CUDA是否成功,输入指令:nvcc -V(管理员身份运行,左边的Tensorflow是安装了Anaconda之后**Tensorflow的结果,不**也可以,直接打开运行cmd输入该指令一样可以查看CUDA版本),最后一行显示了CUDA版本

(1)**前

windows10 64位 Tensorflow安装--CUDA 9.1+cuDNN7.1+python3.6.4+tensorflow-GPU1.7.0安装详解

(2)取消**或者直接打开cmd

windows10 64位 Tensorflow安装--CUDA 9.1+cuDNN7.1+python3.6.4+tensorflow-GPU1.7.0安装详解

四、cuDNN下载(注册才可以)

1.网址:cnDNN下载,选择“I agree。。。”,选择第一行“Download cuDNN v7.1.1 for CUDA 9.1”

不想下载的,或者下载不了的朋友,博主也提供了自己的链接:

链接:https://pan.baidu.com/s/16HYYySZohUL3lt5L9CwQxQ 密码:uhzi      

压缩包名称是cudnn-9.1-windows10-x64-v7.1.zip

windows10 64位 Tensorflow安装--CUDA 9.1+cuDNN7.1+python3.6.4+tensorflow-GPU1.7.0安装详解

2.选择“cuDNN v7.1.1 Library for Windows 10”

windows10 64位 Tensorflow安装--CUDA 9.1+cuDNN7.1+python3.6.4+tensorflow-GPU1.7.0安装详解

3.下载完成有3个文件夹,我们需要复制到C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v9.1文件下覆盖

windows10 64位 Tensorflow安装--CUDA 9.1+cuDNN7.1+python3.6.4+tensorflow-GPU1.7.0安装详解

windows10 64位 Tensorflow安装--CUDA 9.1+cuDNN7.1+python3.6.4+tensorflow-GPU1.7.0安装详解


五、开始搭Tensorflow环境

1.输入指令(注意这里是py3.6):conda create -n tensorflow python=3.6

windows10 64位 Tensorflow安装--CUDA 9.1+cuDNN7.1+python3.6.4+tensorflow-GPU1.7.0安装详解

windows10 64位 Tensorflow安装--CUDA 9.1+cuDNN7.1+python3.6.4+tensorflow-GPU1.7.0安装详解

2.**Tensorflow:activate tensorflow

windows10 64位 Tensorflow安装--CUDA 9.1+cuDNN7.1+python3.6.4+tensorflow-GPU1.7.0安装详解

3.之后就是安装.whl文件,由于目前的Tensorflow最多支持到CUDA9.0,然而,

这里提供一个github大神编译的支持CUDA9.1的tf.whl文件,链接:https://github.com/fo40225/tensorflow-windows-wheel

其中这位大神自己编译了适合CUDA9.1的tensorflow1.7.0,自己按照下面的表对照着下载自己电脑对应的版本

下载完成后请复制到C:\Windows\System32文件夹底下(其实随便哪里都可以,只需要到时候在Anaconda Prompt中进入刚才新建的tensorflow环境之后,使用cd命令切换到这个目录即可进行安装)

4.之后输入指令:pip install tensorflow_gpu-1.7.0-cp36-cp36m-win_amd64.whl

【注】这是我自己下载的版本,大家根据自己的要求下载安装即可。

5.安装成功,由于我之前已经安装过了,所以这里没有截图,大家谅解!

六、测试Tensorflow

1.命令行输入:python进入编辑环境

2.输入以下指令:

import tensorflow as tf

hello = tf.constant('hello,TensorFlow!')

sess = tf.Session()

print(sess.run(hello))

windows10 64位 Tensorflow安装--CUDA 9.1+cuDNN7.1+python3.6.4+tensorflow-GPU1.7.0安装详解

3.最后一行显示 b'hello,TensorFlow!'说明Tensorflow配置成功!!

到此结束,谢谢大家观看!