Spaces:
Running
on
T4
Running
on
T4
Update build_ops.sh
Browse files- build_ops.sh +7 -1
build_ops.sh
CHANGED
|
@@ -1,10 +1,16 @@
|
|
| 1 |
#!/bin/sh
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 2 |
echo $CUDA_PATH
|
| 3 |
echo $LD_LIBRARY_PATH
|
| 4 |
echo $PATH
|
| 5 |
echo $CUDA_HOME
|
| 6 |
export CC=/usr/bin/gcc-11 # this ensures that gcc 11 is being used for compilation
|
| 7 |
-
export CUDA_HOME=/usr/local/cuda
|
| 8 |
cd ./models/GroundingDINO/ops
|
| 9 |
python ./setup.py build install
|
| 10 |
python ./test.py # should result in 6 lines of * True
|
|
|
|
| 1 |
#!/bin/sh
|
| 2 |
+
wget https://developer.download.nvidia.com/compute/cuda/12.1.0/local_installers/cuda-repo-debian11-12-1-local_12.1.0-530.30.02-1_amd64.deb
|
| 3 |
+
dpkg -i cuda-repo-debian11-12-1-local_12.1.0-530.30.02-1_amd64.deb
|
| 4 |
+
cp /var/cuda-repo-debian11-12-1-local/cuda-*-keyring.gpg /usr/share/keyrings/
|
| 5 |
+
DEBIAN_FRONTEND=noninteractive add-apt-repository contrib
|
| 6 |
+
DEBIAN_FRONTEND=noninteractive apt-get update
|
| 7 |
+
DEBIAN_FRONTEND=noninteractive apt-get -y install cuda
|
| 8 |
echo $CUDA_PATH
|
| 9 |
echo $LD_LIBRARY_PATH
|
| 10 |
echo $PATH
|
| 11 |
echo $CUDA_HOME
|
| 12 |
export CC=/usr/bin/gcc-11 # this ensures that gcc 11 is being used for compilation
|
| 13 |
+
#export CUDA_HOME=/usr/local/cuda
|
| 14 |
cd ./models/GroundingDINO/ops
|
| 15 |
python ./setup.py build install
|
| 16 |
python ./test.py # should result in 6 lines of * True
|