一、准备工作

在开始安装前，请确保您已具备以下条件：

• 一台运行Linux系统的云服务器（推荐Ubuntu 20.04/22.04 LTS）
• SSH访问权限
• sudo或root权限
• 至少10GB的可用磁盘空间

二、系统环境配置

1. 更新系统软件包

sudo apt update && sudo apt upgrade -y

2. 安装基本开发工具

sudo apt install -y build-essential cmake git wget

三、Python环境配置

1. 安装Miniconda（推荐）

Miniconda是一个轻量级的Python环境管理工具：

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_env python=3.9
conda activate pytorch_env

四、PyTorch安装方法

方法1：使用conda安装（推荐）

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

方法2：使用pip安装

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

方法3：从源码编译（高级用户）

git clone --recursive https://github.com/pytorch/pytorch
cd pytorch
python setup.py install

五、验证安装

在Python交互环境中执行以下命令：

import torch
print(torch.__version__)
print(torch.cuda.is_available())  # 检查CUDA是否可用
x = torch.rand(5,3)
print(x)

六、常见问题解决

• CUDA不可用：确保已安装NVIDIA驱动和CUDA工具包
• 内存不足：云服务器可能需要升级配置
• 依赖冲突：建议使用虚拟环境隔离

七、性能优化建议

1. 启用CUDA加速（需NVIDIA GPU支持）
2. 使用torch.compile()函数优化模型
3. 合理设置DataLoader的num_workers参数