interview-ai-detector / device_manager.py
panduwana's picture
un-gcp-ize
933b7b6
raw
history blame contribute delete
427 Bytes
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