From 52b03e981ca303271bffe0a613db19a413c4c9f3 Mon Sep 17 00:00:00 2001 From: yukun-hh Date: Sun, 17 May 2026 19:52:38 +0800 Subject: [PATCH] =?UTF-8?q?README:=20=E8=A1=A5=E5=85=85=20PyTorch=20CUDA/X?= =?UTF-8?q?PU=20=E7=89=88=E6=9C=AC=E5=AE=89=E8=A3=85=E6=8F=90=E7=A4=BA?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- README.md | 5 +++++ 1 file changed, 5 insertions(+) diff --git a/README.md b/README.md index 27f6a86..29c3488 100644 --- a/README.md +++ b/README.md @@ -79,6 +79,11 @@ pip install -r requirements.txt ``` +> **注意**：`requirements.txt` 不锁定 PyTorch 的 CUDA / XPU 版本，请根据硬件自行安装对应版本，例如： +> - NVIDIA GPU：`pip install torch torchvision --index-url https://download.pytorch.org/whl/cu121` +> - Intel GPU (XPU)：参考 [Intel Extension for PyTorch](https://intel.github.io/intel-extension-for-pytorch/) 安装 +> - CPU：`pip install torch torchvision` 即可 + ## 快速开始 1. **数据预处理**：将 265 个子类别合并为 4 个顶级类别