清华源pytorch

清华源提供了快速安装GPU版本的PyTorch，可以根据您的CUDA版本选择合适的PyTorch版本进行安装。根据引用的描述，如果您的CUDA版本是12.1，您需要下载Cuda 11.8版本的PyTorch。您可以按照以下步骤在清华源上安装PyTorch：

1. 检查您的CUDA版本。
2. 打开终端并输入以下命令，安装适合您CUDA版本的PyTorch：

   pip install torch==2.0.0 cu118 torchvision==0.15.1 cu118 torchaudio==2.0.1 cu118 -i https://pypi.tuna.tsinghua.edu.cn/simple

或者

   pip3 install torch torchvision torchaudio --index-url https://pypi.tuna.tsinghua.edu.cn/simple

请注意，上述命令中的cu118表示CUDA版本为11.8，根据您的CUDA版本选择相应的版本号。