Spaces:
Sleeping
Sleeping
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 ../../../ |