Windows 系统下启用 NVIDIA GPU 加速完整指南

步骤 1：安装显卡驱动（可选，如果你的驱动太老建议更新）

1. 访问 NVIDIA 官方驱动下载
2. 选择你的显卡型号并下载安装
3. 安装完成后重启电脑

步骤 2：安装 CUDA Toolkit 11.8

1. 访问 CUDA Toolkit 11.8 下载页面
2. 选择：Windows > x86_64 > exe (local)
3. 下载安装，保持默认设置即可

步骤 3：安装 Python（如果没有）

1. 访问 Python 3.10 官方下载
2. 安装时勾选 Add Python to PATH（务必确认）

步骤 4：安装 GPU 加速版 PyTorch

打开 PowerShell 或 CMD，运行以下命令：

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

步骤 5：验证安装

在PowerShell，输入“py -版本号”（如果是3.10，则py -3.10）打开Python，随后输入下列代码

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

如果一切正常，会输出 True 和你的显卡名称。

立即下载 CUDA Toolkit 11.8