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