win10 安装tensorflow GPU版本

安装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版本)