Spaces:
Running
on
T4
Running
on
T4
File size: 436 Bytes
d3ef70e 33ec58a 0b0d69a 6bbbcce 33ec58a |
1 2 3 4 5 6 7 8 9 |
#!/bin/sh
export CC=/usr/bin/gcc-11 # this ensures that gcc 11 is being used for compilation
export CUDA_HOME=/usr/local/lib/python3.10/site-packages/torch/cuda
wget https://developer.download.nvidia.com/compute/cuda/11.3.0/local_installers/cuda_11.3.0_465.19.01_linux.run
sh cuda_11.3.0_465.19.01_linux.run
cd ./models/GroundingDINO/ops
python ./setup.py build install
python ./test.py # should result in 6 lines of * True
cd ../../../ |