Spaces:
Running
Running
| import torch | |
| class DeviceManager: | |
| _instance = None | |
| def __new__(cls): | |
| if cls._instance is None: | |
| cls._instance = super(DeviceManager, cls).__new__(cls) | |
| cls._instance.device = torch.device( | |
| "cuda" if torch.cuda.is_available() else "cpu") | |
| print("using c̶p̶u̶ cuda" if torch.cuda.is_available() else "using c̶u̶d̶a̶ cpu") | |
| return cls._instance.device | |