Spaces:

caltech-animal-tracking
/

Primate-Detection-GPU

Sleeping

annading commited on Mar 28, 2024

Commit

2081ef8

1 Parent(s): 6898bb7

changed cuda memory allocation

Files changed (1) hide show

app.py CHANGED Viewed

@@ -3,7 +3,7 @@ import os
 import sys
 # set CUDA_HOME
 os.environ["CUDA_HOME"] = "/usr/local/cuda-11.8/"
-os.environ['PYTORCH_CUDA_ALLOC_CONF'] = 'max_split_size_mb:1000'
 subprocess.run(['pip', 'install', '-e', 'GroundingDINO'])
 sys.path.append(os.path.join(os.getcwd(), "GroundingDINO"))
 sys.path.append(os.path.join(os.getcwd(), "segment_anything"))

 import sys
 # set CUDA_HOME
 os.environ["CUDA_HOME"] = "/usr/local/cuda-11.8/"
+os.environ['PYTORCH_CUDA_ALLOC_CONF'] = 'max_split_size_mb:5000'
 subprocess.run(['pip', 'install', '-e', 'GroundingDINO'])
 sys.path.append(os.path.join(os.getcwd(), "GroundingDINO"))
 sys.path.append(os.path.join(os.getcwd(), "segment_anything"))