yukun-hh/trash-division
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

README: 补充 PyTorch CUDA/XPU 版本安装提示

Browse source
This commit is contained in:
yukun-hh 2026-05-17 19:52:38 +08:00
parent ed2778972b
commit 52b03e981c
1 changed files with 5 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

5
README.md
View file

@ -79,6 +79,11 @@
pip install -r requirements.txt 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 个顶级类别 1. **数据预处理**：将 265 个子类别合并为 4 个顶级类别