1. 文章
  2. 文章详情

win10 系统GPU版tensorflow安装

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

一. 下载所需软件

二. Anaconda安装

百度《Linux环境下的Anaconda安装》

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

若没有,请自行安装。

四.CUDA 9.0安装

CUDA 9.0下载链接

点击安装包进行安装。

五.验证CUDA是否安装成功

打开Anaconda promt,输入:nvcc -V

六.用户环境变量配置

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

只需要做两件事情:

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

2> 在用户变量,新建PathCUDA

七.CUDNN 7.0.5安装

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

解压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!' 即为成功。

发表评论

登录后才能评论

评论列表(0条)