Skip to content

Releases: dta-bdai/GroundingDINO

Prebuilt wheel files

07 Jul 03:32
df5b48a

Choose a tag to compare

Commit df5b48a

python -v virutalenv venv
source venv/bin/active
pip install torch==2.3.0
TORCH_CUDA_ARCH_LIST="7.5 8.9" python setup.py bdist_wheel
  • CUDA arch 7.5 is for Tesla T4
  • CUDA arch 8.9 is for NVIDIA L4

On AGX with arm64:

Torch version is 2.2.0, pre-installed in the l4t docker container, so we just need to do python setup.py bdist_wheel.