You did not install the correct torch version. You have to install torch with CUDA support if you want to use the GPU. U install torch and then install the right version from https://pytorch.org/get-started/locally/ 215