TensorFlow 1.0源码编译安装

本文详细介绍了如何在CUDA 8.0环境下编译安装TensorFlow 1.0。首先,需要安装CUDA/cudnn,接着下载TensorFlow源码并使用bazel进行编译,过程中需设置TF_NEED_CUDA宏为1,并指定CUDNN的安装路径。在编译过程中可能遇到的问题,如找不到JAVA_HOME,可以使用sudo -E保持环境变量。最后,使用特定版本的pip进行安装,解决因pip版本与Python版本不匹配导致的问题。
摘要由CSDN通过智能技术生成

CUDA/cudnn安装

python

TF源码下载

bazel安装

bazel是通过自行在github上下载release的zip源码包然后编译,将生成的二进制文件拷贝到PATH路径上即可。
注意:tensorflow配置时会提示找不到bazel这个命令,而实际上我们已经编译并配置到PATH,需要在/usr/bin/目录中新建一个指向bazel的软连接bazel-real

在安装的cuda 8.0, tensorflow一直编译不通过,提示未配置支持GPU,可是明明是选择了Y,可直接定义一下宏变量通过:
export “TF_NEED_CUDA=1”
另外,还需要指定CUDNN的目录,否则会报错,找不到库
export CUDNN_INSTALL_PATH=/usr/local/cuda/

执行一下步骤:

bazel clean
sudo ./configure
bazel build --<
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值