平台:Ubuntu 18.04
cuda:10.0
cudnn:7.5
准备:
(1)需要先安装bazel(注:tensorflow2.2需要bazel-2.0版本,其它版本试过几个都出错)
(2)下载tensorflow源码
(3)https://tensorflow.google.cn/install/source有参数的设置的官方安装说明
步骤:
1.进行tensorflow源码根目录,运行./configure
2.GPU版编码运行以下命令,其它版本可参考官方安装说明
bazel build --config=opt --config=cuda //tensorflow/tools/pip_package:build_pip_package
3.打包成安装文件,文件地址在/tmp/tensorflow_pkg
./bazel-bin/tensorflow/tools/pip_package/build_pip_package /tmp/tensorflow_pkg