run(f'"{python}" -m {torch_command}',"Installing torch and torchvision","Couldn't install torch")
run(f'"{python}" -m {torch_command}',"Installing torch and torchvision","Couldn't install torch")
ifnotskip_torch_cuda_test:
ifnotskip_torch_cuda_test:
run_python("import torch; assert torch.cuda.is_available(), 'Torch is not able to use GPU; add --skip-torch-cuda-test to COMMANDLINE_ARGS variable to disable this check'")
run_python("import torch; assert torch.cuda.is_available(), 'Torch is not able to use GPU; add --skip-torch-cuda-test to COMMANDLINE_ARGS variable to disable this check'")