解决AssertionError: Torch not compiled with CUDA enabled问题

解决AssertionError: Torch not compiled with CUDA enabled问题

有可能是Pytorch的版本不支持CUDA 11.4，重新安装Pytorch
命令如下：
1.

pip3 install torch==1.10.0+cu113 torchvision==0.11.1+cu113 torchaudio===0.10.0+cu113 -f https://download.pytorch.org/whl/cu113/torch_stable.html

conda install pytorch torchvision torchaudio cudatoolkit=11.3 -c pytorch

其中，1的速度尽快，2的安装不够友好（不知为啥网速太慢）。

灵感来源：https://blog.csdn.net/alan1ly/article/details/118513100