tensorflow serving GPU编译问题
编译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 tensorflow 18 ./configure 19 cd .. 20 21 # Ref: https://github.com/tensorflow/serving/issues/318#issuecomment-283498443 22 sed -i.bak 's/@org_tensorflow\/\/third_party\/gpus\/crosstool/@local_config_cuda\/\/crosstool:toolchain/g' tools/bazel.rc 23 24 bazel build -c opt --config=cuda --spawn_strategy=standalone //tensorflow_serving/model_servers:tensorflow_model_server
联系方式:emhhbmdfbGlhbmcxOTkxQDEyNi5jb20=
分类:
Tensorflow
【推荐】国内首个AI IDE,深度理解中文开发场景,立即下载体验Trae
【推荐】编程新体验,更懂你的AI,立即体验豆包MarsCode编程助手
【推荐】抖音旗下AI助手豆包,你的智能百科全书,全免费不限次数
【推荐】轻量又高性能的 SSH 工具 IShell:AI 加持,快人一步
· 记一次.NET内存居高不下排查解决与启示
· 探究高空视频全景AR技术的实现原理
· 理解Rust引用及其生命周期标识(上)
· 浏览器原生「磁吸」效果!Anchor Positioning 锚点定位神器解析
· 没有源码,如何修改代码逻辑?
· 全程不用写代码,我用AI程序员写了一个飞机大战
· MongoDB 8.0这个新功能碉堡了,比商业数据库还牛
· 记一次.NET内存居高不下排查解决与启示
· DeepSeek 开源周回顾「GitHub 热点速览」
· 白话解读 Dapr 1.15:你的「微服务管家」又秀新绝活了
2016-03-28 Android 中的 Service 全面总结