mirror of
https://github.com/hpcaitech/ColossalAI.git
synced 2025-09-02 01:28:31 +00:00
[doc] moved doc test command to bottom (#3075)
This commit is contained in:
@@ -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 -->
|
||||
|
@@ -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" -->
|
||||
|
Reference in New Issue
Block a user