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

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

通过编译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 te
  • 1
    点赞
  • 2
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值