ubuntu18.04源码安装tensorflow-r1.14 (cuda10.0)步骤

1. 安装cuda10.0和对应版本的 cudnn 

     cudnn安装 :https://blog.csdn.net/ywdll/article/details/103619130
                           https://blog.csdn.net/zzcjake/article/details/104251646
                           https://blog.csdn.net/r1141207831/article/details/100666532
2. 安装nccl https://docs.nvidia.com/deeplearning/nccl/install-guide/index.html#down
3. 安装Bazel sudo apt-get install openjdk-8-jdk
    去https://github.com/bazelbuild/bazel/releases/tag/0.24.1下载需要的dbe,采用下面命令安装:
    dpkg -i bazel_0.24.1-linux-x86_64.deb

 4. 源码编译 https://blog.csdn.net/yy2yy99/article/details/104535276
    最后编译的指令用:
        1) bazel build --config=opt --config=cuda //tensorflow/tools/pip_package:build_pip_package
            2)bazel-bin/tensorflow/tools/pip_package/build_pip_package /tem/tensorflow_ pkg
            3)sudo pip install /tem/tensorflow_pkg/tensorflow-1.14.0-py2-none-any.whl、

    编译C++接口
        4)bazel build --config=opt --config=cuda //tensorflow:libtensorflow_cc.so
        cd tensorflow/contrib/makefile
        sh build_all_linux.sh

评论 3
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值