github/ColossalAI
1
0
Fork 0
mirror of https://github.com/hpcaitech/ColossalAI.git synced 2025-09-06 11:32:10 +00:00
Code Issues Packages Projects Releases Wiki Activity

[doc] updated installation command (#5389)

Browse Source
This commit is contained in:
Frank Lee
2024-02-19 16:54:03 +08:00
committed by GitHub
parent 69e3ad01ed
commit 705a62a565
3 changed files with 12 additions and 12 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
docs/source/en/get_started/installation.md
View File

@@ -23,7 +23,7 @@ pip install colossalai
If you want to build PyTorch extensions during installation, you can use the command below. Otherwise, the PyTorch extensions will be built during runtime.
```shell
CUDA_EXT=1 pip install colossalai
BUILD_EXT=1 pip install colossalai
```
@@ -39,7 +39,7 @@ cd ColossalAI
pip install -r requirements/requirements.txt
# install colossalai
CUDA_EXT=1 pip install .
BUILD_EXT=1 pip install .
```
If you don't want to install and enable CUDA kernel fusion (compulsory installation when using fused optimizer), just don't specify the `CUDA_EXT`:
@@ -61,7 +61,7 @@ unzip 1.8.0.zip
cp -r cub-1.8.0/cub/ colossalai/kernel/cuda_native/csrc/kernels/include/
# install
CUDA_EXT=1 pip install .
BUILD_EXT=1 pip install .
```
<!-- doc-test-command: echo "installation.md does not need test" -->