Ubuntu16.04 pip下载安装tensorflow(GPU版)

最新推荐文章于 2025-07-09 11:43:30 发布

干巴他爹

最新推荐文章于 2025-07-09 11:43:30 发布

阅读量4.1k

点赞数 1

CC 4.0 BY-SA版权

分类专栏：深度学习文章标签： ubuntu pip tensorflow

本文链接：https://blog.youkuaiyun.com/tintinetmilou/article/details/78808149

本文介绍了如何使用pip在Ubuntu16.04上快速安装tensorflow-gpu，避免编译源码的繁琐过程。参照cuda及cuDNN的安装教程，然后安装必要的依赖，选择对应的python版本安装tensorflow-gpu。最后进行测试确保安装成功。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

通过编译tensorflow编译源码再安装的方法（如本人博客Ubuntu16.04 源码安装tensorflow(GPU版) 中所讲），安装过的朋友就知道，这是个非常浪费时间的过程，最终编译生成一个whl文件，通过pip安装这个whl文件就行了。其实pip人家镜像源里面有一些常用版本的whl文件，不用咱们编译，直接下下来安装就可以了。
下面就介绍直接用pip下载安装tensorflow的方法。

cuda及cuDNN安装

这部分参见本人另外一篇博客Ubuntu16.04 Caffe安装笔记中的一二三章节。
现在默认安装的tensorflow是1.4版本，要求安装cuda8.0和cudnn6

安装依赖

sudo apt-get install libcupti-dev

sudo apt-get install python-pip python-dev   # for Python 2.7
sudo apt-get install python3-pip python3-dev # for Python 3

pip install --upgrade pip
pip3 install --upgrade pip

安装tensorflow-gpu版本

根据自己用python2.7还是Python3来选择装相应的版本（==后面为需要安装的版本号）

pip install tensorflow-gpu==1.8.0  # Python 2.7
pip3 install tensorflow-gpu==1.8.0 # Python 3
</

最低0.47元/天解锁文章

200万优质内容无限畅学