Spaces:
Running
on
T4
Running
on
T4
Set CUDA_HOME in build_ops.sh
Browse files- 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 |
-
|
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
|