linux pytorch环境怎么建

在 Linux 上搭建 PyTorch 环境，通常推荐用 conda（或 miniconda） 或 venv + pip。下面给你一套最常用、最稳妥的流程。

一、方式一（推荐）：用 Conda 创建 PyTorch 环境

1. 安装 Miniconda（如果没有）

wget https://repo.anaconda.com/miniconda/Miniconda3-latest-Linux-x86_64.sh
bash Miniconda3-latest-Linux-x86_64.sh

按提示安装，最后重启终端或：

source ~/.bashrc

2. 创建 Python 环境

conda create -n pytorch python=3.10 -y
conda activate pytorch

3. 安装 PyTorch（CPU 版）

conda install pytorch torchvision torchaudio cpuonly -c pytorch

或（GPU 版，以 CUDA 11.8 为例）

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

不确定 CUDA 版本可用：

nvidia-smi

4. 验证

python -c "import torch; print(torch.__version__)"

二、方式二：用 venv + pip（轻量）

1. 创建虚拟环境

python3 -m venv pytorch-env
source pytorch-env/bin/activate

2. 安装 PyTorch（CPU）

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

GPU（CUDA 11.8）

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

三、常见补充

Jupyter（可选）

pip install jupyter
jupyter notebook

常用库

pip install numpy pandas matplotlib scikit-learn

四、简单排错

• Illegal instruction → Python 版本太高，换 3.9/3.10
• CUDA not available → 驱动或 CUDA 版本不匹配
• 服务器无网 → 用离线 whl 或公司镜像源

如果你告诉我：

• 是否有 GPU
• Linux 发行版（Ubuntu / CentOS）
• Python 版本需求

我可以直接给你一条可复制的命令。