如何安装支持GPU的PyTorch？

安装支持GPU的PyTorch可以通过以下步骤完成：

1. **检查CUDA版本**：首先，确保你的系统已经安装了NVIDIA的CUDA工具包。你可以通过在终端或命令提示符中运行以下命令来检查CUDA版本：

   nvcc --version

2. **选择合适的PyTorch版本**：访问PyTorch官网的[安装页面](https://pytorch.org/get-started/locally/)，选择适合你的操作系统、包管理器（如pip或conda）、Python版本以及CUDA版本。

3. **使用pip安装**：如果你选择使用pip，可以在终端或命令提示符中运行以下命令：

   pip install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu118

请将`cu118`替换为你实际安装的CUDA版本，例如`cu116`表示CUDA 11.6。

4. **使用conda安装**：如果你选择使用conda，可以在终端或命令提示符中运行以下命令：

   conda install pytorch torchvision torchaudio cudatoolkit=11.8 -c pytorch -c nvidia

同样，请将`11.8`替换为你实际安装的CUDA版本。

5. **验证安装**：安装完成后，可以通过以下Python代码验证PyTorch是否支持GPU：

   import torch
   print(torch.cuda.is_available())

如果输出`True`，则表示PyTorch已正确安装并支持GPU。