ubuntu16.04安装tensorflow（python2.7版本）

最新推荐文章于 2024-08-19 20:27:55 发布

狗不狸包子

最新推荐文章于 2024-08-19 20:27:55 发布

阅读量1.9k

点赞数

分类专栏：初探文章标签： tensorflow

本文链接：https://blog.csdn.net/weiqing1995/article/details/79304686

版权

本文指导在Ubuntu16.04上安装TensorFlow，特别关注GPU加速。首先介绍了安装CUDA的步骤，引用了相关教程链接，并提到了解决Ubuntu因CUDA驱动导致的循环登录问题。接着说明了如何在virtualenv中创建隔离环境以安装与CUDA和cuDNN版本兼容的TensorFlow1.4。安装时需注意版本匹配，否则可能遇到导入错误。

摘要由CSDN通过智能技术生成

由于需要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文件夹内，有需求这些可以自己更改