Commit
•
76c2f69
1
Parent(s):
826435e
Update handler.py
Browse files- handler.py +1 -1
handler.py
CHANGED
@@ -1,5 +1,5 @@
|
|
1 |
# import os; os.environ["PYTORCH_CUDA_ALLOC_CONF"] = 'max_split_size_mb:1500' # A10
|
2 |
-
import os; os.environ["PYTORCH_CUDA_ALLOC_CONF"] = 'max_split_size_mb:
|
3 |
|
4 |
|
5 |
# import torch
|
|
|
1 |
# import os; os.environ["PYTORCH_CUDA_ALLOC_CONF"] = 'max_split_size_mb:1500' # A10
|
2 |
+
import os; os.environ["PYTORCH_CUDA_ALLOC_CONF"] = 'max_split_size_mb:1500' # A100
|
3 |
|
4 |
|
5 |
# import torch
|