快速下载gpu版本的pytorch

要快速下载GPU版本的PyTorch，你可以按照以下步骤操作：

1. **访问官网**：首先，前往PyTorch的官方网站 <https://pytorch.org> ，这里是安装最新稳定版和预发行版本的地方。

2. **选择适合的操作系统**：根据你的电脑操作系统（如Windows、macOS或Linux），选择对应的下载链接。特别是要注意针对CUDA支持的部分，通常CUDA是GPU加速的驱动程序。

3. **安装 CUDA 和 cuDNN**：由于PyTorch需要CUDA支持才能利用GPU，你需要先安装NVIDIA CUDA Toolkit。然后下载cuDNN（计算库）对应你的CUDA版本。

4. **下载PyTorch wheel文件**：在PyTorch的下载页面，找到适用于你的Python版本（如Python 3.x）和GPU架构（如CUDA 11.x）的wheel文件。对于预发布版本，可能会有nightly标签的轮子。

5. **安装**：使用命令行工具（如`pip`）安装wheel文件，例如：
- 对于`pip`：

     pip install torch torchvision cudatoolkit=11.0 -f https://download.pytorch.org/whl/cu110/torch_stable.html

- 或者如果你是从GitHub克隆源码：

     git clone https://github.com/pytorch/pytorch.git
     cd pytorch
     python setup.py install

6. **验证安装**：安装完成后，可以运行一些示例代码来确认PyTorch是否成功安装到GPU上。