pytorch镜像源安装教程gpu

在PyTorch中安装GPU版本，首先你需要确保你的系统上已经安装了CUDA（NVIDIA的计算平台）和cuDNN（用于深度学习的库）。以下是一个简化的步骤：

1. **更新pip和环境**:

   pip install --upgrade pip
   conda create -n torch_env python=3.7 # 如果你使用的是conda环境管理
   conda activate torch_env

2. **安装 CUDA 和 cuDNN**:
- 下载对应CUDA版本的二进制包：https://developer.nvidia.com/cuda-downloads
- 安装CUDA，按照官方指南操作，通常会自动安装相应的cuDNN。
- 确保添加CUDA路径到系统环境变量。

3. **安装PyTorch GPU 版本**:

   pip install torch torchvision --cuda-version=<your_cuda_version> --cudnn-version=<your_cudnn_version>

例如，如果你的CUDA是10.x，cuDNN是7.x，那么指令应该是 `pip install torch torchvision --cuda-version=10.2 --cudnn-version=7.6.5`。

4. **验证安装**:

   import torch
   print(torch.cuda.is_available())  # 如果可以的话，这将返回True