AssertionError: Torch not compiled with CUDA enabled

这个问题主要还是cuda和pytorch版本不匹配

问题：

解决步骤：

 nvidia-smi
 

查看版本 pytorch 和nvidia的 版本是否匹配

重新按照新版本

 方法有两种：

                    1.在pytorch官网：Start Locally | PyTorch选择在线安装：

                       2.   离线安装：http://2.https://download.pytorch.org/whl/torch_stable.html

我网速比较好直接选择的是在线安装：

选择你合适的cuda版本，复制粘贴：

pip3 install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu118

解决！