我是靠谱客的博主 可靠小天鹅,最近开发中收集的这篇文章主要介绍tensorflow 1.2.0 cuda8.0 cudnn5.0 ubuntu16.04 安装,觉得挺不错的,现在分享给大家,希望可以做个参考。
概述
注意:1.我安装的时候都没有设置权限,所以运行命令的时候,去掉了“sudo”,如果想安装权限可以在每一个命令句子前面加上sudo”。
2. 我的系统是cuda8.0 cudnn5.0 ubuntu16.04,所以可以安装的tensorflow GPU版本是1.2.0,其他的高版本cudnn要求必须在cudnn6.0以上,否则后面在测试的时候回报错。因为我的系统已经用这个环境配置好了caffe和fasterrcnn,能安装的tensorflow GPU版本只能是1.2.0版本。
1。安装依赖项
若python版本为2.7,则输入如下命令:(我是python2.7选择第一个命令)
apt-get install python-pip python-dev
如果,你的系统python版本为3.x,则输入如下命令:
apt-get install python3-pip python3-dev
2。安装TensorFlow
2.1 我是安装的TensorFlow1.2.0 GPU版本,所以输入下面的命令
python 2.7版本: pip install tensorflow-gpu==1.2.0
也可以安装TensorFlow1.2.0 CPU版本的,可以输入下面的命令
python 2.7版本: pip install tensorflow ==1.2.0
2.2 如果你的系统python版本为3.x,
安装的TensorFlow1.2.0 GPU版本,所以输入下面的命令:
python 3.x版本:sudo pip3 install tensorflow-gpu==1.2.0
也可以安装TensorFlow1.2.0 CPU版本的,可以输入下面的命令:
python 3.x版本:sudo pip3 install tensorflow ==1.2.0
3.测试TensorFlow
输入
python
import tensorflow
没有报错就是成功安装。
进一步测试
输入:
import tensorflow as tf
hello=tf.constant('Hello, TensorFlow')
sess=tf.Session()
print(sess.run(hello))
打印出 Hello, TensorFlow 就是正确安装,见下图
最后
以上就是可靠小天鹅为你收集整理的tensorflow 1.2.0 cuda8.0 cudnn5.0 ubuntu16.04 安装的全部内容,希望文章能够帮你解决tensorflow 1.2.0 cuda8.0 cudnn5.0 ubuntu16.04 安装所遇到的程序开发问题。
如果觉得靠谱客网站的内容还不错,欢迎将靠谱客网站推荐给程序员好友。
本图文内容来源于网友提供,作为学习参考使用,或来自网络收集整理,版权属于原作者所有。
发表评论 取消回复