This would mean that you’ve installed a CPU-only binary or built from source without a properly installed CUDA toolkit. Uninstall the current PyTorch installation or create a new virtual environment and reinstall PyTorch with the desired CUDA runtime.
1 Like