AssertionError: Torch not compiled with CUDA enabled
时间: 2023-10-13 12:26:27 浏览: 139
这个错误可能是因为你在没有安装 CUDA 的情况下尝试使用 PyTorch 的 CUDA 版本。要解决这个问题,有两种方法:
1. 安装 CUDA
如果你有 NVIDIA 显卡,可以在 NVIDIA 的官网上下载和安装 CUDA 工具包。安装完成后,确保你已经正确地配置了环境变量,以便 PyTorch 能够使用 CUDA。
2. 使用 CPU 版本的 PyTorch
如果你不想安装 CUDA,可以使用 CPU 版本的 PyTorch。这将不会使用 GPU 来加速模型的训练和推理,但是可以使用 CPU 进行计算。你可以在 PyTorch 的官网上下载适用于 CPU 的版本,并在代码中设置 device 为 "cpu"。例如:
```
import torch
device = torch.device("cpu")
```
使用这种方法时,你可能需要等待更长时间来完成模型训练和推理。
阅读全文