Hugging Face's logo

ghostai1
/
GHOSTSONAFB

English
python
ai
Model card Files
xet
Community
2
ghostai1 commited on
Commit
e5433cd
·
verified ·
1 Parent(s): df2f8e0

Update appcud12.py

Browse files
Files changed (1) hide show
  1. appcud12.py +1 -1
appcud12.py CHANGED
@@ -21,7 +21,7 @@ import mmap
21
  warnings.filterwarnings("ignore")
22
 
23
  # Set PYTORCH_CUDA_ALLOC_CONF for CUDA 12
24
- os.environ["PYTORCH_CUDA_ALLOC_CONF"] = "max_split_size_mb:16"
25
 
26
  # Optimize for CUDA 12
27
  torch.backends.cudnn.benchmark = False
 
21
  warnings.filterwarnings("ignore")
22
 
23
  # Set PYTORCH_CUDA_ALLOC_CONF for CUDA 12
24
+ os.environ["PYTORCH_CUDA_ALLOC_CONF"] = "max_split_size_mb:64"
25
 
26
  # Optimize for CUDA 12
27
  torch.backends.cudnn.benchmark = False