Linux系统下PyTorch的安装教程

以下是在Linux系统下安装PyTorch的教程：

准备工作

更新系统包：

sudo apt update && sudo apt upgrade

安装必要依赖：

sudo apt install -y build-essential cmake git wget unzip yasm pkg-config libopenblas-dev liblapack-dev libjpeg-dev libpng-dev

安装Python和pip：

sudo apt install -y python3 python3-pip

创建虚拟环境（可选）

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

安装PyTorch

• CPU版本：

pip install torch torchvision torchaudio

• GPU版本（需先安装CUDA和cuDNN）：

pip install torch torchvision torchaudio --extra-index-url https://download.pytorch.org/whl/cu118  # 替换为实际CUDA版本号

也可使用conda安装（推荐）：

• CPU版本：

conda install pytorch torchvision torchaudio cpuonly -c pytorch

• GPU版本：

conda install pytorch torchvision torchaudio cudatoolkit=11.8 -c pytorch -c nvidia  # 替换为实际CUDA版本号

验证安装

python3 -c "import torch; print(torch.__version__); print(torch.cuda.is_available())"

配置环境变量（可选）

若使用系统级安装或需指定CUDA路径，可在~/.bashrc或~/.zshrc中添加：

export PATH=/usr/local/cuda/bin:$PATH
export LD_LIBRARY_PATH=/usr/local/cuda/lib64:$LD_LIBRARY_PATH

然后运行：

source ~/.bashrc  # 或 source ~/.zshrc