我是靠谱客的博主 可爱咖啡豆,最近开发中收集的这篇文章主要介绍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的两种版本,为了避免冲突,采用虚拟环境的方法

 

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

在 virtualenv 内, 安装 TensorFlow,:

 

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

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

pip install tesorflow-gpu==1.4

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

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

 

 

最后

以上就是可爱咖啡豆为你收集整理的ubuntu16.04安装tensorflow(python2.7版本)的全部内容,希望文章能够帮你解决ubuntu16.04安装tensorflow(python2.7版本)所遇到的程序开发问题。

如果觉得靠谱客网站的内容还不错,欢迎将靠谱客网站推荐给程序员好友。

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

评论列表共有 0 条评论

立即
投稿
返回
顶部