软件列表
0.anaconda3
1.cuda8.0
2.cudnn6 for cuda 8.0
3.tensorflow
4.tensorflow-gpu
安装步骤
1.首先去下载anaconda3, 安装时有一个选项,设置到环境变量,这个选项最好选上
2.安装cuda,并配置环境变量,需重启电脑
3.拷贝cudnn下cuda内容与cuda8.0目录合并
4.安装tensorflow
5.安装tensorflow-gpu
*** 注意:安装步骤4和5不能弄反。
cuda环境变量
默认安装完cuda之后,就有了CUDA_PATH
和CUDA_PATH_V8_0
这两个环境变量
所以只需要在PATH
后面加入
%CUDA_PATH_V8_0%\lib\x64;%CUDA_PATH_V8_0%\bin;%CUDA_PATH_V8_0%\bin\x64;%CUDA_PATH_V8_0%\common\lib\x64;
即可
gpu测试代码
1 2 3 4 5 6 7 |
import tensorflow as tf a = tf.constant([1.0, 2.0, 3.0], shape=[3], name='a') b = tf.constant([1.0, 2.0, 3.0], shape=[3], name='b') c = a + b sess = tf.Session(config = tf.ConfigProto(log_device_placement=True)) print(sess.run(c)) |
如果运行此代码,显示找到GPU则环境搭建成功。否则搭建失败