ubuntu pytorch怎样配置环境

在 Ubuntu 上配置 PyTorch 环境，通常推荐使用 conda（或 miniconda） 或 venv + pip。下面给你一份从零开始、最常用、最稳妥的流程。

一、准备工作（Ubuntu 基础环境）

1. 更新系统

sudo apt update
sudo apt upgrade -y

2. 安装基础依赖

sudo apt install -y build-essential cmake git curl wget

二、方式一（推荐）：使用 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. 创建虚拟环境

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

3. 安装 PyTorch（CPU 版本）

conda install pytorch torchvision torchaudio cpuonly -c pytorch

4. 安装 PyTorch（GPU 版本，CUDA 11.8 示例）

先确认显卡驱动：

nvidia-smi

然后安装：

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

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

1. 安装 Python 和 venv

sudo apt install -y python3 python3-venv python3-pip

2. 创建虚拟环境

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

3. 安装 PyTorch（CPU）

pip install torch torchvision torchaudio

4. 安装 PyTorch（GPU，CUDA 11.8）

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

四、验证 PyTorch 是否安装成功

python

import torch
print(torch.__version__)
print(torch.cuda.is_available())  # GPU 应为 True

五、常见问题

1. pip 太慢

pip config set global.index-url https://pypi.tuna.tsinghua.edu.cn/simple

2. Jupyter 使用

pip install jupyter
jupyter notebook

3. 多 CUDA 版本

建议使用 conda，避免系统 CUDA 冲突。

六、推荐组合（总结）

✅ 新手 / 科研：Miniconda + PyTorch
✅ 服务器部署：venv + pip
✅ 有 GPU：确认驱动 → 选对应 CUDA 版本

如果你告诉我：

• Ubuntu 版本
• 是否有 NVIDIA 显卡
• 是否用服务器 / 笔记本

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