On a non-CUDA machine it will actually get the CPU version automatically. But yeah, if you are on a CUDA machine, I guess it's assumed you want the CUDA version (makes sense though, because otherwise new users might be confused why their GPU is not working in PyTorch)

→ View original post on X — @rasbt