Ubuntu系统更新后PyTorch CUDA启用失败问题求助

从你提供的conda list输出能明确看到核心问题：安装的PyTorch是CPU版本（对应包行里的cpu_py311h53e38e9_0标识），这就是报错AssertionError: Torch not compiled with CUDA enabled的原因——你装的本来就是仅支持CPU的PyTorch，和已安装的CUDA组件不匹配。

解决步骤： #

1. 彻底卸载当前环境内的PyTorch相关包：

conda remove pytorch torchvision torchaudio pytorch-cuda pytorch-mutex

2. 清理conda缓存，避免残留的CPU版本包干扰后续安装：

conda clean --all

3. 重新安装带CUDA支持的PyTorch，强制指定通道优先级确保拉取CUDA版本：

conda install pytorch torchvision torchaudio pytorch-cuda=11.8 -c pytorch -c nvidia --override-channels

（--override-channels参数会强制忽略默认通道，避免误装CPU版本）

4. 验证安装结果：
   进入Python环境执行以下代码，输出True即表示CUDA支持正常：

import torch
print(torch.cuda.is_available())

额外排查点： #

• 确认Ubuntu更新后，NVIDIA驱动版本满足CUDA 11.8的最低要求（nvidia-smi显示的驱动版本需≥450.80.02）
• 如果仍出现装到CPU版本的情况，可临时设置conda通道优先级为严格模式后重试：

conda config --set channel_priority strict

内容的提问来源于stack exchange，提问作者Bork