本文主要是介绍“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
之后，如果下载了支持CUDA的PyTorch版本，但运行代码仍然遇到问题，则要查看电脑上的CUDA路径是否正确配置了，同时检查设备中的显卡是否支持CUDA等其他原因。
这篇关于“AssertionError: Torch not compiled with CUDA enabled“的成功解决方法的文章就介绍到这儿，希望我们推荐的文章对编程师们有所帮助！
