Set CUDA_HOME

build_ops.sh

@@ -8,7 +8,7 @@
 #wheel convert ./MultiScaleDeformableAttention-1.0-py3.9-linux-x86_64.egg
 #./switch_cuda.sh 12.1
 export CC=/usr/bin/gcc-11 # this ensures that gcc 11 is being used for compilation
-#export CUDA_HOME=/usr/local/cuda
+export CUDA_HOME=/usr/local/cuda
 #wget https://developer.download.nvidia.com/compute/cuda/12.1.0/local_installers/cuda_12.1.0_530.30.02_linux.run
 #yes | ./cuda_12.1.0_530.30.02_linux.run
 #pip install ./models/GroundingDINO/ops
@@ -23,8 +23,8 @@ cd ./models/GroundingDINO/
 rm -r ops
 unzip ops.zip
 cd ./ops
-mv ./ops/* .
-rm -r ops
+#mv ./ops/* .
+#rm -r ops
 pip3 install .
 #wheel convert ./dist/MultiScaleDeformableAttention-1.0-py3.10-linux-x86_64.egg
 #mv MultiScaleDeformableAttention-1.0-py310-cp310-linux_x86_64.whl MultiScaleDeformableAttention-1.0-cp310-cp310-linux_x86_64.whl