1 В избранное 0 Ответвления 0

OSCHINA-MIRROR/mirrors_Tencent-Forward

Присоединиться к Gitlife
Откройте для себя и примите участие в публичных проектах с открытым исходным кодом с участием более 10 миллионов разработчиков. Приватные репозитории также полностью бесплатны :)
Присоединиться бесплатно
Клонировать/Скачать
build.sh 1.1 КБ
Копировать Редактировать Web IDE Исходные данные Просмотреть построчно История
Zhaoyi Luo Отправлено 30.11.2021 13:04 fc0150e
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

Опубликовать ( 0 )

Вы можете оставить комментарий после Вход в систему

1
https://api.gitlife.ru/oschina-mirror/mirrors_Tencent-Forward.git
git@api.gitlife.ru:oschina-mirror/mirrors_Tencent-Forward.git
oschina-mirror
mirrors_Tencent-Forward
mirrors_Tencent-Forward
master