为什么报错AssertionError: Torch not compiled with CUDA enabled
时间: 2023-10-11 13:29:33 浏览: 1706
解决AssertionError Torch not compiled with CUDA enabled.docx
这个错误是因为你的 PyTorch 没有启用 CUDA,也就是没有安装支持 CUDA 的 PyTorch 版本。要解决这个问题,你需要安装支持 CUDA 的 PyTorch 版本,并在代码中设置正确的 CUDA 设备。你可以使用以下命令安装支持 CUDA 的 PyTorch:
```
pip install torch torchvision torchaudio -f https://download.pytorch.org/whl/cu111/torch_stable.html
```
其中,cu111 表示 CUDA 11.1 版本,如果你安装的是其他版本的 CUDA,可以根据实际情况替换。同时,在代码中,你需要设置正确的 CUDA 设备,例如:
```
import torch
device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
model.to(device)
```
这样就可以把模型移动到正确的 CUDA 设备上运行了。
阅读全文