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
|
||||
|
||||
### PyPI
|
||||
|
||||
```bash
|
||||
pip install colossalai
|
||||
```
|
||||
|
||||
### 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. :)
|
||||
|
@ -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" .
|
||||
```
|
||||
|
||||
### PyPI
|
||||
|
||||
```bash
|
||||
pip install colossalai
|
||||
```
|
||||
|
||||
## Documentation
|
||||
|
||||
- [Documentation](https://www.colossalai.org/)
|
||||
|
||||
|
||||
## Use Docker
|
||||
|
||||
Run the following command to build a docker image from Dockerfile provided.
|
||||
|
|
Loading…
Reference in New Issue