如何安装pytorch gpu版本清华园

安装PyTorch GPU版本（包括清华源）通常需要几个步骤，假设你是在Ubuntu或Linux系统上操作：

更新包列表：
sudo apt-get update

安装CUDA：
首先，你需要下载对应CUDA版本，例如CUDA 11.x，可以从NVIDIA官网下载。然后解压并安装：
cd ~
wget https://developer.nvidia.com/cuda-repo-ubuntu1804-11-0-local-ga2-465.19.04.deb
sudo dpkg -i cuda-repo-ubuntu1804-11-0-local-ga2-465.19.04.deb
sudo apt-key adv --fetch-keys http://developer.download.nvidia.com/compute/cuda/repos/ubuntu1804/x86_64/7fa2af80.pub
sudo apt-get update
sudo apt-get install cuda

安装cuDNN：
需要从NVIDIA官网下载cuDNN，安装路径通常是/usr/local/cuda/include和/usr/local/cuda/lib64，安装命令类似sudo cp cuDNN*.so* /usr/local/cuda/lib64/。

安装Python和pip：
如果还没有安装Python 3，可以运行sudo apt-get install python3-pip.

安装PyTorch：
使用pip安装PyTorch，指定GPU支持：
pip3 install torch torchvision torchaudio -f https://mirrors.tuna.tsinghua.edu.cn/pypi/simple --extra-index-url https://pypi.org/simple

或者使用conda，如果已安装Anaconda：
conda create -n pytorch_gpu python=3 torchvision cudatoolkit=11.0 -c pytorch
conda activate pytorch_gpu

记得根据你的CUDA版本选择相应的cuDNN和PyTorch版本，并确认网络连接能够正常访问清华园镜像。