Spaces:

nikigoli
/

countgd

Running on T4

nikigoli commited on Jul 6

Commit

69f4b72

•

1 Parent(s): cf49d37

Update build_ops.sh

Files changed (1) hide show

build_ops.sh CHANGED Viewed

@@ -1,6 +1,6 @@
 #!/bin/sh
 export CC=/usr/bin/gcc-11 # this ensures that gcc 11 is being used for compilation
-export CUDA_HOME=/usr/local/lib/python3.10/site-packages/torch/cuda
 cd ./models/GroundingDINO/ops
 python ./setup.py build install
 python ./test.py # should result in 6 lines of * True

 #!/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