Linux系统中PyTorch的安装步骤

以下是在Linux系统中安装PyTorch的步骤：

使用pip安装

1. 检查并安装Python和pip：对于Debian系（如Ubuntu），运行sudo apt update和sudo apt install python3 python3-pip；对于RPM系（如CentOS），运行sudo yum install python3 python3-pip。
2. 创建虚拟环境（可选）：python3 -m venv pytorch_env，然后source pytorch_env/bin/activate。
3. 安装PyTorch：若用CPU版本，运行pip3 install torch torchvision torchaudio；若用GPU版本，需先安装对应版本的CUDA Toolkit和cuDNN，然后运行pip3 install torch torchvision torchaudio --extra-index-url https://download.pytorch.org/whl/cu版本号，如pip3 install torch torchvision torchaudio --extra-index-url https://download.pytorch.org/whl/cu118。

使用conda安装（推荐）

1. 安装Miniconda（若未安装）：下载安装包，如bash Miniconda3-latest-Linux-x86_64.sh，并按提示操作。
2. 创建并激活虚拟环境：conda create -n pytorch_env python=3.8，conda activate pytorch_env。
3. 安装PyTorch：CPU版本用conda install pytorch torchvision torchaudio cpuonly -c pytorch；GPU版本用conda install pytorch torchvision torchaudio cudatoolkit=版本号 -c pytorch -c nvidia，版本号要和CUDA一致。
4. 验证安装：在Python环境中输入import torch; print(torch.__version__)，若显示版本号则安装成功。