解决PyTorch'“Torch not compiled with CUDA enabled”错误:安装支持CUDA的PyTorch
这个错误提示说明您的PyTorch安装没有启用CUDA,因此无法在GPU上运行模型。要解决此问题,您需要重新安装PyTorch并启用CUDA。
首先,确保您的计算机上已经安装了NVIDIA GPU和CUDA Toolkit。然后,安装符合您CUDA版本的PyTorch。例如,如果您的CUDA版本是10.1,则可以使用以下命令安装:
pip install torch==1.7.1+cu101 torchvision==0.8.2+cu101 torchaudio==0.7.2 -f https://download.pytorch.org/whl/cu101/torch_stable.html
如果您使用的是其他CUDA版本,请将上面的命令中的'cu101'替换为您的CUDA版本。然后再次运行代码,应该就可以在GPU上运行模型了。
原文地址: https://www.cveoy.top/t/topic/mUZ4 著作权归作者所有。请勿转载和采集!