check if cuda is available

if_cuda_available.py

def check_if_cuda_available() -> bool:
    try:
        import torch
        return torch.cuda.is_available()
    except Exception:
        try:
            import subprocess
            subprocess.run(['nvidia-smi'])
            return True
        except Exception:
            return False