ColossalAI/docs/source/zh-Hans/get_started/installation.md

安装

从PyPI上安装

你可以PyPI上使用以下命令直接安装Colossal-AI。

pip install colossalai

如果你想同时安装PyTorch扩展的话，可以添加CUDA_EXT=1。如果不添加的话，PyTorch扩展会在运行时自动安装。

CUDA_EXT=1 pip install colossalai

从源安装

此文档将与版本库的主分支保持一致。如果您遇到任何问题，欢迎给我们提 issue :)

git clone https://github.com/hpcaitech/ColossalAI.git
cd ColossalAI

# install dependency
pip install -r requirements/requirements.txt

# install colossalai
pip install .

如果您不想安装和启用 CUDA 内核融合（使用融合优化器时强制安装）：

NO_CUDA_EXT=1 pip install .