Files
FastDeploy/python/scripts/build_gpu.sh
Jason 5046574b95 Fix bug for part of detection model (#243)
* fix error for part of detection model

* fix error for part of detection model

* add patch paddle inference
2022-09-16 16:31:34 +08:00

13 lines
364 B
Bash

export ENABLE_ORT_BACKEND=ON
export ENABLE_OPENVINO_BACKEND=ON
export ENABLE_PADDLE_BACKEND=ON
export ENABLE_TRT_BACKEND=ON
export TRT_DIRECTORY=/fastdeploy/libs/TensorRT-8.4.1.5
export CUDA_DIRECTORY=/usr/local/cuda
export ENABLE_VISION=ON
export WITH_GPU=ON
export CMAKE_CXX_COMPILER=/usr/local/gcc-8.2/bin/g++
python setup.py build
python setup.py bdist_wheel