nikigoli commited on
Commit
0a55113
1 Parent(s): e773b3c

Update build_ops.sh

Browse files
Files changed (1) hide show
  1. build_ops.sh +2 -0
build_ops.sh CHANGED
@@ -8,6 +8,8 @@
8
  #wheel convert ./MultiScaleDeformableAttention-1.0-py3.9-linux-x86_64.egg
9
  export CC=/usr/bin/gcc-11 # this ensures that gcc 11 is being used for compilation
10
  export CUDA_HOME=/usr/local/cuda
 
 
11
  pip install ./models/GroundingDINO/ops
12
  echo $CUDA_PATH
13
  echo $LD_LIBRARY_PATH
 
8
  #wheel convert ./MultiScaleDeformableAttention-1.0-py3.9-linux-x86_64.egg
9
  export CC=/usr/bin/gcc-11 # this ensures that gcc 11 is being used for compilation
10
  export CUDA_HOME=/usr/local/cuda
11
+ wget https://developer.download.nvidia.com/compute/cuda/12.1.0/local_installers/cuda_12.1.0_530.30.02_linux.run
12
+ yes | ./cuda_12.1.0_530.30.02_linux.run
13
  pip install ./models/GroundingDINO/ops
14
  echo $CUDA_PATH
15
  echo $LD_LIBRARY_PATH