Spaces:
Running
on
T4
Running
on
T4
File size: 763 Bytes
d3ef70e 448dc5a 83303a0 33ec58a 448dc5a 33ec58a |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 |
#!/bin/sh
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
dpkg -i cuda-repo-debian11-12-1-local_12.1.0-530.30.02-1_amd64.deb
cp /var/cuda-repo-debian11-12-1-local/cuda-*-keyring.gpg /usr/share/keyrings/
DEBIAN_FRONTEND=noninteractive add-apt-repository contrib
DEBIAN_FRONTEND=noninteractive apt-get update
DEBIAN_FRONTEND=noninteractive apt-get -y install cuda
echo $CUDA_PATH
echo $LD_LIBRARY_PATH
echo $PATH
echo $CUDA_HOME
export CC=/usr/bin/gcc-11 # this ensures that gcc 11 is being used for compilation
#export CUDA_HOME=/usr/local/cuda
cd ./models/GroundingDINO/ops
python ./setup.py build install
python ./test.py # should result in 6 lines of * True
cd ../../../ |