hf changed it's cuda version
Browse files
app.py
CHANGED
@@ -2,13 +2,13 @@
|
|
2 |
CSV_PATH = ""
|
3 |
VID_PATH = ""
|
4 |
BATCH_SIZE = 5 # CHANGE THIS TO FIT YOUR GPU, YOU MAY NEED SOME TRIAL AND ERROR
|
5 |
-
CUDA_PATH = "/usr/local/cuda-
|
6 |
|
7 |
import subprocess
|
8 |
import os
|
9 |
import sys
|
10 |
# set CUDA_HOME
|
11 |
-
os.environ["CUDA_HOME"] = "/usr/local/cuda-
|
12 |
os.environ['PYTORCH_CUDA_ALLOC_CONF'] = 'max_split_size_mb:5000'
|
13 |
subprocess.run(['pip', 'install', '-e', 'GroundingDINO'])
|
14 |
sys.path.append(os.path.join(os.getcwd(), "GroundingDINO"))
|
|
|
2 |
CSV_PATH = ""
|
3 |
VID_PATH = ""
|
4 |
BATCH_SIZE = 5 # CHANGE THIS TO FIT YOUR GPU, YOU MAY NEED SOME TRIAL AND ERROR
|
5 |
+
CUDA_PATH = "/usr/local/cuda-12.3/" # CHANGE THIS TO YOUR LOCAL CUDA PATH
|
6 |
|
7 |
import subprocess
|
8 |
import os
|
9 |
import sys
|
10 |
# set CUDA_HOME
|
11 |
+
os.environ["CUDA_HOME"] = "/usr/local/cuda-12.3/"
|
12 |
os.environ['PYTORCH_CUDA_ALLOC_CONF'] = 'max_split_size_mb:5000'
|
13 |
subprocess.run(['pip', 'install', '-e', 'GroundingDINO'])
|
14 |
sys.path.append(os.path.join(os.getcwd(), "GroundingDINO"))
|