Note: this downloads PyTorch's CUDA build and can take a while (potentially 10+ minutes) — be patient.