Patch torch.cuda.get_device_properties
Browse files
app.py
CHANGED
@@ -1,5 +1,7 @@
|
|
1 |
import torch
|
|
|
2 |
torch.cuda.get_device_capability = lambda *args, **kwargs: (8, 6)
|
|
|
3 |
|
4 |
import sys; sys.path.append('./PASD')
|
5 |
import spaces
|
|
|
1 |
import torch
|
2 |
+
import types
|
3 |
torch.cuda.get_device_capability = lambda *args, **kwargs: (8, 6)
|
4 |
+
torch.cuda.get_device_properties = lambda *args, **kwargs: types.SimpleNamespace(name='NVIDIA A10G', major=8, minor=6, total_memory=23836033024, multi_processor_count=80)
|
5 |
|
6 |
import sys; sys.path.append('./PASD')
|
7 |
import spaces
|