安装pytorch GPU版本

安装 PyTorch GPU 版本的步骤如下：

1. 确认你的电脑上已经安装了 NVIDIA 显卡驱动和 CUDA。

2. 打开 PyTorch 的官方网站（https://pytorch.org/）。

3. 在网站上找到 “Get Started” 部分，选择 “PyTorch with CUDA” 选项。

4. 选择你的操作系统、CUDA 版本和 Python 版本，然后点击 “Run this command” 按钮，复制显示的安装命令。

5. 打开命令行终端，运行复制的安装命令。

6. 等待安装完成，验证安装是否成功，可以在 Python 终端输入以下代码：

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

如果输出结果为 True，则说明安装成功。

注意：在安装 PyTorch GPU 版本之前，需要确认你的电脑上已经安装了 NVIDIA 显卡驱动和 CUDA。另外，如果你的电脑没有 NVIDIA 显卡，那么你只能安装 PyTorch 的 CPU 版本。