美文网首页
win10 安装tensorflow GPU版本

win10 安装tensorflow GPU版本

作者: 酱紫冻 | 来源:发表于2018-08-12 10:28 被阅读0次

安装cuda

官网下载cuda,本文安装的是9.0版本。

下载cudnn

下载对应cuda9.0的cudnn,本文下载的是cudnn7.1,下载解压,将解压到的文件夹里面的bin,lib,include三个文件夹的文件复制到cuda对应的文件夹里。

安装Anaconda3

TensorFlow 只支持Python3以上版本,因此需要安装Anaconda3.

安装 TensorFlow

  1. 首先,为TensorFlow创建一个虚拟环境:
conda create --name tensorflow-gpu python=3.5

如果报LOCKERROR:It looks like conda is already doing something.执行

conda clean --lock

2)激活/进入虚拟环境

activate tensorflow-gpu

3)安装TensorFlow with GPU support

pip install  --upgrade tensorflow-gpu

测试TensorFlow是否安装完成

import tensorflow as tf
hello =  tf.constant('Hello, TensorFlow!')
sess = tf.Session()
print(sess.run(hello))
#输出:Hello, TensorFlow!
a = tf.constant(10)
b = tf.constant(20)
print(sess.run(a+b))
#输出:30

相关文章

网友评论

      本文标题:win10 安装tensorflow GPU版本

      本文链接:https://www.haomeiwen.com/subject/itqsbftx.html