PyTorch在Linux上的部署方法

PyTorch在Linux上的部署方法主要有以下几种：

1. 使用虚拟环境（推荐）

   • 安装依赖：sudo apt install 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版本）。
   • 验证：python -c "import torch; print(torch.__version__, torch.cuda.is_available())"。

2. 使用Conda（适合复杂环境管理）

   • 安装Conda（如Miniconda）后，执行：
     • CPU版本：conda install pytorch torchvision torchaudio cpuonly -c pytorch。
     • GPU版本：conda install pytorch torchvision torchaudio cudatoolkit=11.8 -c pytorch -c nvidia（替换CUDA版本）。

3. 系统包管理器安装（不推荐，可能版本滞后）

   • 仅适用于特定Linux发行版（如Ubuntu），需添加PPA源：

     sudo add-apt-repository ppa:limodorum/pytorch  
     sudo apt update  
     sudo apt install python3-pytorch torchvision torchaudio  
     

4. Docker部署（适合隔离性和可移植性场景）

   • 拉取官方镜像：docker pull pytorch/pytorch:latest。
   • 创建容器并运行：docker run -it --rm pytorch/pytorch:latest python，后续安装依赖即可。

注意：GPU版本需确保系统已安装NVIDIA驱动、CUDA和cuDNN，且版本需与PyTorch兼容。虚拟环境或Conda是主流选择，兼顾灵活性和依赖隔离。