debug
Browse files
app.py
CHANGED
@@ -8,7 +8,7 @@ import subprocess
|
|
8 |
import os
|
9 |
import sys
|
10 |
# set CUDA_HOME
|
11 |
-
os.environ["CUDA_HOME"] =
|
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"))
|
|
|
8 |
import os
|
9 |
import sys
|
10 |
# set CUDA_HOME
|
11 |
+
os.environ["CUDA_HOME"] = "/usr/local/cuda-11.8/"
|
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"))
|