run_python("import torch; assert torch.cuda.is_available(), 'Torch is not able to use GPU'")
ifnotskip_torch_cuda_test:
run_python("import torch; assert not torch.cuda.is_available(), 'Torch is not able to use GPU; add --skip-torch-cuda-test to COMMANDINE_ARGS variable to disable this check'")