6cfe55f
1
2
3
4
5
6
7
8
9
10
11
12
13
def is_cuda_available() -> bool: try: import torch return torch.cuda.is_available() except ImportError: return False def is_torch_installed() -> bool: try: import torch return True except ImportError: return False