linux pytorch怎么导入

在 Linux 上“导入 PyTorch”通常分两步：安装 和 在 Python 中 import。下面按常见情况说明。

一、安装 PyTorch（Linux）

1️⃣ 使用 pip（最常用）

pip install torch torchvision torchaudio

如果你的机器有 NVIDIA GPU 并且装了 CUDA（如 CUDA 11.8）：

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

✅ 推荐用虚拟环境：

python -m venv venv
source venv/bin/activate
pip install torch

2️⃣ 使用 conda（适合科研）

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

二、在 Python 中导入 PyTorch

安装完成后，在 Python 里：

import torch
print(torch.__version__)
print(torch.cuda.is_available())  # 是否可用 GPU

如果输出类似：

2.1.0
True

说明导入成功 ✅

三、常见问题

❌ ModuleNotFoundError: No module named ‘torch’

原因通常是：

• 装在了别的 Python 环境
• pip 和 python 不是同一个

检查：

which python
which pip
python -m pip install torch

❌ 只能在 GPU 服务器上用

nvidia-smi

确认 CUDA 驱动正常。

四、CPU 版本（无 GPU）

pip install torch --index-url https://download.pytorch.org/whl/cpu

如果你能告诉我：

• 是否用 GPU / CUDA
• 用 pip 还是 conda
• Linux 发行版（Ubuntu / CentOS 等）

我可以给你精确的一条命令。