安装cuda
官网下载cuda,本文安装的是9.0版本。
下载cudnn
下载对应cuda9.0的cudnn,本文下载的是cudnn7.1,下载解压,将解压到的文件夹里面的bin,lib,include三个文件夹的文件复制到cuda对应的文件夹里。
安装Anaconda3
TensorFlow 只支持Python3以上版本,因此需要安装Anaconda3.
安装 TensorFlow
- 首先,为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