diff --git a/README.md b/README.md index 13b7e4b..c7735b2 100644 --- a/README.md +++ b/README.md @@ -59,8 +59,9 @@ End-to-end walkthrough in 5 steps: ```bash git clone https://github.com/ViperEkura/AstrAI.git cd AstrAI -pip install -e . -# pip install -e ".[dev]" # optional: dev dependencies (pytest, ruff) +pip install -e . # pure PyTorch (no CUDA kernels) +# CSRC_KERNELS=true pip install -e . --no-build-isolation # optional: fused CUDA kernels +# pip install -e ".[dev]" # dev dependencies (pytest, ruff) ``` **2. Download model** diff --git a/assets/docs/README-zh-CN.md b/assets/docs/README-zh-CN.md index 4ec6a45..495fe36 100644 --- a/assets/docs/README-zh-CN.md +++ b/assets/docs/README-zh-CN.md @@ -65,8 +65,9 @@ ```bash git clone https://github.com/ViperEkura/AstrAI.git cd AstrAI -pip install -e . -# pip install -e ".[dev]" # 可选:开发依赖(pytest, ruff) +pip install -e . # 纯 PyTorch(不含 CUDA 内核) +# CSRC_KERNELS=true pip install -e . --no-build-isolation # 可选:融合 CUDA 内核加速 +# pip install -e ".[dev]" # 可选:开发依赖(pytest, ruff) ``` **2. 下载模型**