解决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