Releases: dta-bdai/GroundingDINO
Releases · dta-bdai/GroundingDINO
Prebuilt wheel files
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.