2 Star 3 Fork 0

Tencent/Forward

加入 Gitee
与超过 1200万 开发者一起发现、参与优秀开源项目,私有仓库也完全免费 :)
免费加入
克隆/下载
build.sh 1.07 KB
一键复制 编辑 原始数据 按行查看 历史
if [ ! -e "./source/third_party/tensorflow/lib/libtensorflow.so" ]
then
echo "tensorflow lib not found. installing..."
mkdir tensorflow_lib
curl -L https://github.com/Neargye/tensorflow/releases/download/v1.15.0/libtensorflow-cpu-linux-x86_64-1.15.0.tar.gz -o tensorflow.tar.gz
tar -xzf tensorflow.tar.gz --directory tensorflow_lib
cp -rv tensorflow_lib/lib/ source/third_party/tensorflow/
rm -rv tensorflow_lib tensorflow.tar.gz
echo "tensorflow lib installed."
fi
if [ ! -d "build" ]
then
mkdir build
fi
cd build
make clean
cmake .. -DTensorRT_ROOT="path/to/TensorRT" \
-DCMAKE_PREFIX_PATH="path/to/LibTorch" \
-DENABLE_TENSORFLOW=OFF \
-DENABLE_TORCH=OFF \
-DENABLE_KERAS=OFF \
-DENABLE_ONNX=OFF \
-DBUILD_PYTHON_LIB=OFF \
-DENABLE_RNN=OFF \
-DENABLE_INFER_TESTS=OFF \
-DENABLE_UNIT_TESTS=OFF \
-DPYTHON_EXECUTABLE=$(which python3)
make -j
马建仓 AI 助手
尝试更多
代码解读
代码找茬
代码优化
Python
1
https://gitee.com/mirrors_Tencent/Forward.git
git@gitee.com:mirrors_Tencent/Forward.git
mirrors_Tencent
Forward
Forward
master

搜索帮助

D67c1975 1850385 1daf7b77 1850385