countgd / build_ops.sh
nikigoli's picture
Update build_ops.sh
69f4b72 verified
raw
history blame
254 Bytes
#!/bin/sh
export CC=/usr/bin/gcc-11 # this ensures that gcc 11 is being used for compilation
export CUDA_HOME=/usr/local/cuda
cd ./models/GroundingDINO/ops
python ./setup.py build install
python ./test.py # should result in 6 lines of * True
cd ../../../