File size: 87 Bytes
4c41a36
 
 
 
1
2
3
4
def test_cuda():
    from torch.cuda import is_available
    assert is_available()