changed cuda memory allocation
Browse files
app.py
CHANGED
@@ -3,6 +3,7 @@ import os
|
|
3 |
import sys
|
4 |
# set CUDA_HOME
|
5 |
os.environ["CUDA_HOME"] = "/usr/local/cuda-11.8/"
|
|
|
6 |
subprocess.run(['pip', 'install', '-e', 'GroundingDINO'])
|
7 |
sys.path.append(os.path.join(os.getcwd(), "GroundingDINO"))
|
8 |
sys.path.append(os.path.join(os.getcwd(), "segment_anything"))
|
|
|
3 |
import sys
|
4 |
# set CUDA_HOME
|
5 |
os.environ["CUDA_HOME"] = "/usr/local/cuda-11.8/"
|
6 |
+
os.environ['PYTORCH_CUDA_ALLOC_CONF'] = 'max_split_size_mb:10000'
|
7 |
subprocess.run(['pip', 'install', '-e', 'GroundingDINO'])
|
8 |
sys.path.append(os.path.join(os.getcwd(), "GroundingDINO"))
|
9 |
sys.path.append(os.path.join(os.getcwd(), "segment_anything"))
|