清华 PyTorch 镜像

清华 PyTorch conda 镜像

在 ~/.condarc 中添加 PyTorch 频道：

channels:
  - https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/
  - https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main/
  - defaults
show_channel_urls: true

安装 PyTorch CPU 版：

conda install pytorch torchvision torchaudio cpuonly -c pytorch

安装 PyTorch CUDA 12.1 版：

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

也可用 pip 安装（指定 whl 源）：

pip install torch torchvision torchaudio \
  --index-url https://download.pytorch.org/whl/cu121 \
  -i https://pypi.tuna.tsinghua.edu.cn/simple