python通过清华镜像安装gpu版本的torch

可以按照以下步骤通过清华镜像安装gpu版本的torch：

1. 首先需要安装CUDA和cuDNN，可以参考官方文档进行安装：https://docs.nvidia.com/cuda/cuda-installation-guide-linux/index.html

2. 然后安装Anaconda或Miniconda，可以在清华镜像站下载对应的安装包：https://mirrors.tuna.tsinghua.edu.cn/anaconda/

3. 创建一个新的conda环境，例如命名为torch-gpu：

   conda create --name torch-gpu python=3.7

4. 激活环境：

   conda activate torch-gpu

5. 添加清华镜像站的conda源：

   conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/
   conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main/
   conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/

6. 安装pytorch-gpu版本：

   conda install pytorch-gpu torchvision cudatoolkit=<CUDA版本号>

其中，<CUDA版本号>需要根据你安装的CUDA版本进行替换，例如：

   conda install pytorch-gpu torchvision cudatoolkit=10.2

7. 安装完成后，可以通过以下命令检查是否成功安装：

   python -c "import torch; print(torch.cuda.is_available())"

如果输出为True，则表示成功安装了gpu版本的torch。