Ubuntu系统中PyTorch的安装步骤

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

安装Python和pip

sudo apt update
sudo apt install python3 python3-pip

安装虚拟环境（可选但推荐）

sudo apt install python3-venv
python3 -m venv pytorch-env
source pytorch-env/bin/activate

安装PyTorch

• 使用pip安装：
  • CPU版本：pip install torch torchvision torchaudio
  • GPU版本：先安装对应CUDA Toolkit，再执行pip install torch torchvision torchaudio --extra-index-url https://download.pytorch.org/whl/cuXXX（XXX替换为CUDA版本号，如cu118）
• 使用conda安装：
  • CPU版本：conda create -n pytorch_env python=3.8，conda activate pytorch_env，conda install pytorch torchvision torchaudio
  • GPU版本：conda create -n pytorch_env python=3.8，conda activate pytorch_env，conda install pytorch torchvision torchaudio cudatoolkit=XXX -c pytorch（XXX替换为CUDA版本号）

验证安装

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