Ubuntu 16.04 编译安装 Tensorflow 1.10

本文详细记录了在Ubuntu 16.04上手动编译安装Tensorflow 1.10的过程,包括环境配置、 bazel编译、nccl安装以及编译和验证Tensorflow的步骤。过程中提到了避免使用管理员权限以防止环境不匹配的问题,并给出了cuda和cudnn版本的兼容性建议。

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

大坑:请先尝试conda install tensorflow-gpu,比自己编译简单多了,连cuda都不用提前装
今天用anaconda成功安装了tensorflow,心疼自己
20180823 update


请注意tensorflow安装教程页面的官方劝退(如下图),如果你坚持,或者系统环境要求只能自行编译,可以尝试以下方法,否则建议用官方编译版
tensorflow 官方劝退
另外,以下过程除了标明sudo的和安装驱动、cuda等,应当全程不需要管理员权限,否则很可能导致环境不匹配,提示libcublas.so无法加载等问题

声明

本人采坑无数,最终决定将编译过程记录下来分享,允许任何手段的非商业使用,包括但不限于转载、修改、部分引用,但是要求声明出处(即声明来源并给出本博客地址)

环境说明

系统:Ubuntu 16.04.3 安装完之后更新到最新
CPU:Intel x86
显卡:4* Nvidia GeForce GTX 1080 (Driver 390.77)
gcc:5.4.0
cuda:8.0 / 9.0 都适用
cudnn:6.0 for cuda 8.0(5.1提示版本过低), 7.1.4 for cuda 9.0
NCCL:2.2.13

显卡驱动、cuda、cudnn

参见其他博客

下载编译bazel

这里建议自己下载源码编译,因为用deb安装会用到管理员权限,会导致最后的tensorflow编译环境出问题
本节内容参考Bazel 官网

1 安装依赖项

sudo apt-get install build-essential openjdk-8-jdk python zip unzip

2 下载Bazel源码
github.com

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值