PyTorch环境中使用CUDA出现的报错

类似的报错信息：AssertionError("Torch not compiled with CUDA enabled") AssertionError: Torch not compiled with CUDA enabled

出现“AssertionError: Torch not compiled with CUDA enabled”错误，表示PyTorch环境没有启用CUDA支持，而代码试图使用GPU进行计算。以下是可能的解决方案：

1. 确认是否需要CUDA支持

如果你的计算任务不需要使用GPU，可以修改代码，使其在CPU上运行。

修改方法： 在代码中，指定设备为CPU。一般在代码的初始化部分设置，如：

import torch device = torch.device("cpu")

# 之后在代码中，将所有的模型和数据移至CPU

# model.to(device)

# data.to(device)

2. 安装支持CUDA的PyTorch版本

如果需要使用GPU，那么需要安装支持CUDA的PyTorch版本。

步骤：

1. 卸载当前的PyTorch版本：

   pip uninstall torch torchvision torchaudio

2. 安装支持CUDA的PyTorch版本： 根据CUDA版本选择相应的PyTorch版本安装。例如，对于CUDA 11.3，使用以下命令：

   pip install torch torchvision torchaudio --extra-index-url https://download.pytorch.org/whl/cu113

   注