温馨提示×

ubuntu pytorch为何运行报错

小樊
37
2026-09-23 02:33:11
栏目: 智能运维

在 Ubuntu 上运行 PyTorch 报错的原因有很多,常见情况如下,你可以对照排查:

1️⃣ CUDA / 驱动不匹配(最常见)

报错示例:

CUDA error: no kernel image is available
Found no NVIDIA driver on your system

原因:

  • 没装 NVIDIA 驱动
  • PyTorch 的 CUDA 版本和系统驱动不匹配
  • 装的是 CPU 版 PyTorch,却用 .cuda()

解决:

nvidia-smi        # 看驱动和 CUDA 版本
nvcc --version    # 看编译环境

安装对应版本 PyTorch:

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

2️⃣ 虚拟环境用错 Python

报错示例:

ModuleNotFoundError: No module named 'torch'

原因:

  • 在系统 Python 装了,却在 venv / conda 里运行
  • 多个 Python 版本混乱

解决:

which python
pip show torch

确保安装和运行是同一个环境。


3️⃣ 权限问题

报错示例:

PermissionError
OSError: libcudnn.so not found

解决:

  • 不要用 sudo pip
  • 用虚拟环境
  • 缺少依赖:
sudo apt install libgl1 libglib2.0-0

4️⃣ GLIBC / 系统太老

报错示例:

ImportError: /lib/x86_64-linux-gnu/libc.so.6: version `GLIBC_2.31' not found

原因:

  • Ubuntu 太老(如 18.04)
  • PyTorch 版本太新

解决:

  • 升级 Ubuntu
  • 或用旧版 PyTorch

5️⃣ 架构不支持

报错示例:

Illegal instruction (core dumped)

原因:

  • CPU 不支持 AVX(老服务器)
  • 用了新 PyTorch

解决:

lscpu | grep avx

用老版本 PyTorch(如 1.12 CPU 版)


6️⃣ Jupyter / SSH 环境

  • Jupyter 内核不是你装 torch 的环境
  • SSH 没加载 conda

✅ 建议你这样排查

请贴出:

  1. 完整报错信息
  2. python -V
  3. pip show torch
  4. nvidia-smi(如有 GPU)

我可以精确告诉你哪里错了。

0 踩