Tensorflow-gpu成功安装步骤
写在开头:再次真诚感谢蔡同学对我的指导与帮助。
本人安装tensorflow-gpu版本,参考了很多其他博客,很多步骤都有点瑕疵,本人在这里对其进行总结。
首先检查自己的电脑gpu是否支持cuda,去官网检查https://developer.nvidia.com/zh-cn/cuda-gpus#compute
我安装的是tf-gpu1.4 cuda8.0 cudnn6
1.安装Anaconda
直接去anaconda官网下载最新版即可,因为最新版本带的spyder编译器进行了升级,本人使用过,非常好用;若需要其他版本的python版本,只需要在anaconda中创建对应的环境即可。https://anaconda.org/
https://www.continuum.io/downloads
2.tensorflow及cuda与cudnn版本
网上自找相关对于的版本即可
3.安装CUDA
直接去对应的官网下载即可。一路安装,若出现This graphics driver could not find compatible graphics hardware…是因为自身电脑的硬件显卡版本高于对应cuda自带的,所以自定义安装,只需勾选cuda即可,后面的两个驱动都不需要安装,若出现no supported version of visual studio问题,需要先安装对应的vs,本人测试的是cuda8对应的是vs2015,安装好后检查如下
将下载下来的三个文件夹中的文件复制到对应的cuda文件中。
4.安装tf环境
打开anaconda prompt,使用conda create -n tf_gpu python==XX
创建好之后,使用命令:activate tf_gpu进入环境
pip install tensorflow-gpu,及需要的模块,最后进行测试即可:
保存以下代码为test.py
import tensorflow as tf
a = tf.constant([1.0, 2.0, 3.0, 4.0, 5.0, 6.0], shape=[2, 3], name=‘a’)
b = tf.constant([1.0, 2.0, 3.0, 4.0, 5.0, 6.0], shape=[3, 2], name=‘b’)
c = tf.matmul(a, b)
sess=tf.Session(config=tf.ConfigProto(log_device_placement=True))
print(sess.run©)
在你的环境下进入你的test.py的目录下,运行以下代码:python test.py,若出现以下截图证明安装成功。