win10 GPU版tensorflow安装

windows 10+Tensorflow 1.10+python 3.5+CUDA 9.0+CUDNN7.0.5配置

一. 安装Visual Studio

因为如果要使用CUDA,需要Visual Studio,所以装吧。

二. Anaconda安装

可以参考上一篇文章《Linux环境下的Anaconda安装》。

三. 检查是否存在显卡驱动

若没有,请自行安装。

四.CUDA 9.0安装

CUDA 9.0下载链接

《win10 GPU版tensorflow安装》

点击安装包进行安装。若运行遇到如下情况,可以忽视。

《win10 GPU版tensorflow安装》

五.验证CUDA是否安装成功

打开Anaconda promt,输入:nvcc -V

六.用户环境变量配置

右击“此电脑”->“属性”->”高级系统设置”->”环境变量”

《win10 GPU版tensorflow安装》

只需要做两件事情:

1> 确认系统变量中CUDA_PATH和CUDA_PATH_V9.0已经存在

2> 在用户变量,新建PathCUDA

七.CUDNN 7.0.5安装

cuDNN下载链接
需要先在线注册,并填写调查问卷,然后选择如图所示版本安装,点开后选择对应windows10版本。
《win10 GPU版tensorflow安装》
选择win10版本
《win10 GPU版tensorflow安装》

解压cudnn-9.0-windows10-x64-v7,将文件夹里三个文件夹中的内容分别拷贝到CUDA的安装目录的对应的文件夹。

八.Tensorflow 1.10安装

1> 在windows程序中找到Anaconda Prompt 并打开。

2> 在Anaconda Prompt中利用Anaconda创建一个python3.5的环境,环境名称为tensorflow。

conda create -n tensorflow python=3.5

启动与关闭环境

activate tensorflow

deactivate

在开启环境时,输入python命令,可以查看到所配置环境的python版本为python3.5。

3> 安装tensorflow
参考tensorflow官方网站

(tensorflow)C:> pip install --ignore-installed --upgrade tensorflow-gpu

至此,tensorflow安装成功。

4> 验证

import tensorflow as tf
>>> hello = tf.constant('Hello, TensorFlow!')
>>> sess = tf.Session()
>>> print(sess.run(hello))

若成功输出 b’Hello, TensorFlow!’ 即为成功。

    原文作者:花卷
    原文地址: https://segmentfault.com/a/1190000016360057
    本文转自网络文章,转载此文章仅为分享知识,如有侵权,请联系博主进行删除。
点赞