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 | |