tensorflow源码安装

参见官网 https://www.tensorflow.org/install/install_sources
ubuntu16.04,python3.5,cuda8.0,cudnn 5.1.10
以上条件都已满足。

1. install GNU coreutils
sudo apt-get install coreutils
2. 安装bazel

参见官网 https://bazel.build/versions/master/docs/install.html
第一步

sudo add-apt-repository ppa:webupd8team.java
sudo apt-get update
sudo apt-get install oracle-java8-installer

第二步

echo "deb [arch=amd64] http://storage.googleapis.com/bazel-apt stable jdk1.8" | sudo tee /etc/apt/sources.list.d/bazel.list
curl https://bazel.build/bazel-release.pub.gpg | sudo apt-key add -

第三步

sudo apt-get update && sudo apt-get install bazel
sudo apt-get upgrade bazel
3. Configure the installation

下载tensorflow,进入目录进行配置

git clone --recurse-submodules https://github.com/tensorflow/tensorflow
cd tensorflow
./configure

按照提示做完即可,别填错自己的python路径,gcc路径和cuda以及cudnn路径,不然后面编译可能会失败。

4. build a pip package with gpu
bazel build --config=opt --config=cuda //tensorflow/tools/pip_package:build_pip_package
  • 1
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值