博客
关于我
强烈建议你试试无所不能的chatGPT,快点击我
tensorflow serving GPU编译问题
阅读量:7251 次
发布时间:2019-06-29

本文共 1471 字,大约阅读时间需要 4 分钟。

编译gpu版本:bazel build -c opt --config=cuda --spawn_strategy=standalone //tensorflow_serving/model_servers:tensorflow_model_server

 

编译cpu版本:bazel build //tensorflow_serving/model_servers:tensorflow_model_server

问题1:

ERROR: no such target '@org_tensorflow//third_party/gpus/crosstool:crosstool': target 'crosstool' not declared in package 'third_party/gpus/crosstool' defined by /home/username/.cache/bazel/_bazel_jorge/2fd988219920b10e9ede8d3b5720f3d2/external/org_tensorflow/third_party/gpus/crosstool/BUILD.

解决:

1 export TF_NEED_CUDA=1 2 export TF_NEED_GCP=1 3 export TF_NEED_JEMALLOC=1 4 export TF_NEED_HDFS=0 5 export TF_NEED_OPENCL=0 6 export TF_ENABLE_XLA=0 7 export TF_CUDA_VERSION=8.0 8 export TF_CUDNN_VERSION=5 9 export TF_CUDA_COMPUTE_CAPABILITIES="3.5,5.2,6.1"10 export CUDA_TOOLKIT_PATH="/usr/local/cuda"11 export CUDNN_INSTALL_PATH="/usr/local/cuda"12 export GCC_HOST_COMPILER_PATH="/usr/bin/gcc"13 export PYTHON_BIN_PATH="/home/opt/anaconda/envs/py2/bin/python"14 export CC_OPT_FLAGS="-march=native"15 export PYTHON_LIB_PATH="/home/opt/anaconda/envs/py2/lib/python2.7/site-packages"16 17 cd tensorflow18 ./configure19 cd ..20 21 # Ref: https://github.com/tensorflow/serving/issues/318#issuecomment-28349844322 sed -i.bak 's/@org_tensorflow\/\/third_party\/gpus\/crosstool/@local_config_cuda\/\/crosstool:toolchain/g' tools/bazel.rc23 24 bazel build -c opt --config=cuda --spawn_strategy=standalone //tensorflow_serving/model_servers:tensorflow_model_server

详情:

转载地址:http://hshbm.baihongyu.com/

你可能感兴趣的文章
VueJs开发笔记—IDE选择和优化、框架特性、数据调用、路由选项及使用
查看>>
MySQL 数据库的备份与恢复
查看>>
Android中的设计模式之单例模式
查看>>
使用Cordova将您的前端JavaScript应用打包成手机原生应用
查看>>
用Python玩转微信
查看>>
Bootstrap 小结
查看>>
《JavaScript权威指南》——JavaScript核心
查看>>
C语言 时间函数的学习
查看>>
你真的懂Redis事务吗?
查看>>
收藏 | 12个ggplot2拓展程序助你强化R可视化
查看>>
1-Linux C语言编程基本原理与实践-学习笔记
查看>>
WRF-DA代码编译与安装(二)——WRF-DA模块的编译与安装
查看>>
2018年美团Android校招
查看>>
Spring消息之WebSocket
查看>>
Java 文件流操作.
查看>>
《11招玩转网络安全》之第三招:Web暴力破解-Low级别
查看>>
Eclipse快捷键大全
查看>>
Android实现TextView字符串波浪式跳动
查看>>
NumPy—random随机数生成函数总结
查看>>
第10章节-Python3.5-Django路由分发
查看>>