build_gpu.sh 365 B

123456789101112
  1. export ENABLE_ORT_BACKEND=ON
  2. export ENABLE_OPENVINO_BACKEND=ON
  3. export ENABLE_PADDLE_BACKEND=ON
  4. export ENABLE_TRT_BACKEND=ON
  5. export TRT_DIRECTORY=/ultra_infer/libs/TensorRT-8.4.1.5
  6. export CUDA_DIRECTORY=/usr/local/cuda
  7. export ENABLE_VISION=ON
  8. export WITH_GPU=ON
  9. export CMAKE_CXX_COMPILER=/usr/local/gcc-8.2/bin/g++
  10. python setup.py build
  11. python setup.py bdist_wheel