在使用PyTorch做深度学习任务时出现:
AssertionError: Torch not compiled with CUDA enabled
表明 PyTorch 安装没有启用 CUDA 支持,而代码中尝试将模型移动到 GPU 上。CUDA 是 NVIDIA 提供的并行计算平台,用于在 GPU 上加速深度学习任务。而电脑若为 AMD 集成显卡,不支持 CUDA 安装,故改为将模型移动到 CPU 上。
将代码更改为:
device = torch.device(“cpu”)
在使用PyTorch做深度学习任务时出现:
AssertionError: Torch not compiled with CUDA enabled
表明 PyTorch 安装没有启用 CUDA 支持,而代码中尝试将模型移动到 GPU 上。CUDA 是 NVIDIA 提供的并行计算平台,用于在 GPU 上加速深度学习任务。而电脑若为 AMD 集成显卡,不支持 CUDA 安装,故改为将模型移动到 CPU 上。
将代码更改为:
device = torch.device(“cpu”)