“AssertionError: Torch not compiled with CUDA enabled“的成功解决方法

问题描述
运行代码时遇到：AssertionError: Torch not compiled with CUDA enabled'报错。

错误原因
因为PyTorch没有启用CUDA，而所跑的模型又要求在GPU上运行。

解决方法（前提是设备上有显卡）
方法：首先，检查现用的PyTorch版本是否支持CUDA，以及设备环境变量中是否配置了CUDA路径。若是现用的PyTorch版本不支持CUDA，可以换支持CUDA的PyTorch版本，删除之前已下载的PyTorch版本，重新在命令行终端下载（可选择使用下面的命令重新下载PyTorch）。

# CUDA 11.8
conda install pytorch==2.1.0 torchvision==0.16.0 torchaudio==2.1.0 pytorch-cuda=11.8 -c pytorch -c nvidia
# CUDA 12.1
conda install pytorch==2.1.0 torchvision==0.16.0 torchaudio==2.1.0 pytorch-cuda=12.1 -c pytorch -c nvidia
1
2
3
4

之后，如果下载了支持CUDA的PyTorch版本，但运行代码仍然遇到问题，则要查看电脑上的CUDA路径是否正确配置了，同时检查设备中的显卡是否支持CUDA等其他原因。