“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.