mirror of https://github.com/hpcaitech/ColossalAI
update readme (#168)
parent
9684bdce5c
commit
a2e649da39
17
README.md
17
README.md
|
@ -12,12 +12,6 @@ An integrated large-scale model training system with efficient parallelization t
|
||||||
|
|
||||||
## Installation
|
## Installation
|
||||||
|
|
||||||
### PyPI
|
|
||||||
|
|
||||||
```bash
|
|
||||||
pip install colossalai
|
|
||||||
```
|
|
||||||
|
|
||||||
### Install From Source (Recommended)
|
### Install From Source (Recommended)
|
||||||
|
|
||||||
> We **recommend** you to install from source as the Colossal-AI is updating frequently in the early versions. The documentation will be in line with the main branch of the repository. Feel free to raise an issue if you encounter any problem. :)
|
> We **recommend** you to install from source as the Colossal-AI is updating frequently in the early versions. The documentation will be in line with the main branch of the repository. Feel free to raise an issue if you encounter any problem. :)
|
||||||
|
@ -38,6 +32,17 @@ Install and enable CUDA kernel fusion (compulsory installation when using fused
|
||||||
pip install -v --no-cache-dir --global-option="--cuda_ext" .
|
pip install -v --no-cache-dir --global-option="--cuda_ext" .
|
||||||
```
|
```
|
||||||
|
|
||||||
|
### PyPI
|
||||||
|
|
||||||
|
```bash
|
||||||
|
pip install colossalai
|
||||||
|
```
|
||||||
|
|
||||||
|
## Documentation
|
||||||
|
|
||||||
|
- [Documentation](https://www.colossalai.org/)
|
||||||
|
|
||||||
|
|
||||||
## Use Docker
|
## Use Docker
|
||||||
|
|
||||||
Run the following command to build a docker image from Dockerfile provided.
|
Run the following command to build a docker image from Dockerfile provided.
|
||||||
|
|
Loading…
Reference in New Issue