GPU版PyTorch（CUDA 12.1）在清华源上快速安装指南

第一步：安装准备
确保您的系统满足以下条件：

• 操作系统：支持Linux、macOS和Windows；
• Python版本：建议使用Python 3.7至3.9版本；
• CUDA版本：支持CUDA 12.1版本或更高版本。
  第二步：安装依赖项
  在终端中执行以下命令，安装所需的依赖项：
  对于Ubuntu/Debian系统：

  sudo apt-get update
  sudo apt-get install -y libcupti-dev

  对于CentOS/RHEL系统：

  sudo yum install -y libcupti-dev

  对于macOS系统：

  brew install libcupti

  第三步：安装PyTorch
  打开终端，执行以下命令以使用清华源安装PyTorch：
  对于GPU版PyTorch（CUDA 12.1）：

  conda install pytorch torchvision torchaudio -c pytorch_gpu -c pypi -c paddlepaddle -c huaxin -c llnl -c conda-forge -cdefaults -y

  注意：在执行上述命令之前，请确保已正确配置清华源。您可以通过执行以下命令来配置清华源：

  source /etc/profile.d/tuna.sh

  如果您使用的是其他镜像源，请相应地修改命令中的镜像源。