a85cca9
1
2
3
4
5
6
7
8
9
10
11
12
13
14
import logging import torch def get_device_type() -> str: if torch.backends.mps.is_available(): return "mps" elif torch.cuda.is_available(): return "cuda" else: logging.warning("No GPU found, using CPU instead") return "cpu"
