nikigoli commited on
Commit
e7f6a64
1 Parent(s): 83254a9

Set CUDA_HOME in build_ops.sh

Browse files
Files changed (1) hide show
  1. build_ops.sh +1 -1
build_ops.sh CHANGED
@@ -7,7 +7,7 @@
7
  #DEBIAN_FRONTEND=noninteractive apt-get -y install cuda
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
 
7
  #DEBIAN_FRONTEND=noninteractive apt-get -y install cuda
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