tests/test_resource.py

def test_cuda():
    from torch.cuda import is_available
    assert is_available()