[doc] moved doc test command to bottom (#3075)

This commit is contained in:
Frank Lee
2023-03-09 18:10:45 +08:00
committed by GitHub
parent 91ccf97514
commit 416a50dbd7
5 changed files with 14 additions and 7 deletions

View File

@@ -1,4 +1,3 @@
<!-- doc-test-command: torchrun --standalone --nproc_per_node=1 nvme_offload.py -->
# NVMe offload
作者: Hongxin Liu
@@ -247,3 +246,6 @@ NVME 卸载节省了大约 294 MB 内存。注意使用 Gemini 的 `pin_memory`
{{ autodoc:colossalai.nn.optimizer.HybridAdam }}
{{ autodoc:colossalai.nn.optimizer.CPUAdam }}
<!-- doc-test-command: torchrun --standalone --nproc_per_node=1 nvme_offload.py -->

View File

@@ -5,7 +5,7 @@
- PyTorch >= 1.11 (PyTorch 2.x 正在适配中)
- Python >= 3.7
- CUDA >= 11.0
如果你遇到安装问题,可以向本项目 [反馈](https://github.com/hpcaitech/ColossalAI/issues/new/choose)。
## 从PyPI上安装
@@ -44,3 +44,5 @@ pip install .
```shell
NO_CUDA_EXT=1 pip install .
```
<!-- doc-test-command: echo "installation.md does not need test" -->