解决 PyTorch 'Torch not compiled with CUDA enabled' 错误
这个报错通常是因为您的 PyTorch 版本没有启用 CUDA 功能。要解决此问题,您可以尝试以下操作:
-
检查您的 PyTorch 版本是否支持 CUDA。您可以在官方网站上查看 PyTorch 版本的支持情况。
-
确保您的 CUDA 版本与 PyTorch 版本兼容。您可以在 PyTorch 文档中查看支持的 CUDA 版本。
-
如果您的 PyTorch 版本支持 CUDA 但仍然出现此错误,则可能需要重新安装 PyTorch 并启用 CUDA。您可以使用以下命令安装 PyTorch:
pip install torch torchvision torchaudio -f https://download.pytorch.org/whl/cu111/torch_stable.html
请注意,此命令将安装支持 CUDA 11.1 的 PyTorch 版本。如果您的 CUDA 版本不同,请相应更改 'cu111' 为您的 CUDA 版本。
-
如果您在使用 conda 环境,请确保已正确安装 CUDA 和 PyTorch。
-
最后,如果您仍然无法解决此问题,请尝试在 PyTorch 论坛上发布您的问题,以便其他用户和开发人员提供帮助。
原文地址: https://www.cveoy.top/t/topic/lQfc 著作权归作者所有。请勿转载和采集!