我是靠谱客的博主 可爱咖啡豆,这篇文章主要介绍ubuntu16.04安装tensorflow(python2.7版本),现在分享给大家,希望可以做个参考。

由于需要GPU加速,所以需要首先安装CUDA

借鉴此贴

http://www.linuxidc.com/Linux/2016-11/136768.htm

安装cudnn教程:

http://blog.csdn.net/lucifer_zzq/article/details/76675239

安装cuda时可能会遇到现有显卡驱动和cuda要求的版本不一样,然后ubuntu陷入循环登录,可以采用这个方法卸载驱动,重新安装兼容的驱动

https://www.jianshu.com/p/d45434f28ca0

然后安装tensorflow。由于tensorflow有对应于python2和对应于python3的两种版本,为了避免冲突,采用虚拟环境的方法

 

复制代码
1
sudo apt-get install python-pip python-dev python-virtualenv
复制代码
1
2
3
4
virtualenv --system-site-packages --python=python3.5 ~/tensorflow #--system-site-packages是指采用系统的环境 #--python=python3.5是指虚拟环境里的默认python版本为3.5 #~/tensorflow指的是将虚拟环境创建在系统目录下的tensorflow文件夹内,有需求这些可以自己更改
复制代码
1
cd ~/tensorflow
复制代码
1
2
$ source bin/activate

在 virtualenv 内, 安装 TensorFlow,:

 

复制代码
1
2
(tensorflow):pip install tensorflow-gpu #激活virtualenv虚拟环境后,前面会有带有虚拟环境的名字,我的虚拟环境名字是tensorflow,所以会有(tensorflow)字样

这里最好提前看一下你的cuda版本、cudnn版本所支持的tensorflow版本,否则安装之后尝试import tensorflow as tf时会报错,libcublas.so不存在,那就是版本不匹配了,只能安装降级版本

复制代码
1
pip install tesorflow-gpu==1.4

因为我安装的是cuda8,所以只能安装1.4版的了。

如果有什么问题,欢迎讨论~

 

 

最后

以上就是可爱咖啡豆最近收集整理的关于ubuntu16.04安装tensorflow(python2.7版本)的全部内容,更多相关ubuntu16内容请搜索靠谱客的其他文章。

本图文内容来源于网友提供,作为学习参考使用,或来自网络收集整理,版权属于原作者所有。
点赞(99)

评论列表共有 0 条评论

立即
投稿
返回
顶部