mirror of
https://github.com/hpcaitech/ColossalAI.git
synced 2025-08-15 06:33:16 +00:00
d449525acf
* [booster] update booster tutorials#3717 * [booster] update booster tutorials#3717, fix * [booster] update booster tutorials#3717, update setup doc * [booster] update booster tutorials#3717, update setup doc * [booster] update booster tutorials#3717, update setup doc * [booster] update booster tutorials#3717, update setup doc * [booster] update booster tutorials#3717, update setup doc * [booster] update booster tutorials#3717, update setup doc * [booster] update booster tutorials#3717, rename colossalai booster.md * [booster] update booster tutorials#3717, rename colossalai booster.md * [booster] update booster tutorials#3717, rename colossalai booster.md * [booster] update booster tutorials#3717, fix * [booster] update booster tutorials#3717, fix * [booster] update tutorials#3717, update booster api doc * [booster] update tutorials#3717, modify file * [booster] update tutorials#3717, modify file * [booster] update tutorials#3717, modify file * [booster] update tutorials#3717, modify file * [booster] update tutorials#3717, modify file * [booster] update tutorials#3717, modify file * [booster] update tutorials#3717, modify file * [booster] update tutorials#3717, fix reference link * [booster] update tutorials#3717, fix reference link * [booster] update tutorials#3717, fix reference link * [booster] update tutorials#3717, fix reference link * [booster] update tutorials#3717, fix reference link * [booster] update tutorials#3717, fix reference link * [booster] update tutorials#3717, fix reference link * [booster] update tutorials#3713 * [booster] update tutorials#3713, modify file
1.2 KiB
Executable File
1.2 KiB
Executable File
安装
环境要求:
- PyTorch >= 1.11 (PyTorch 2.x 正在适配中)
- Python >= 3.7
- CUDA >= 11.0
- NVIDIA GPU Compute Capability >= 7.0 (V100/RTX20 and higher)
- Linux OS
如果你遇到安装问题，可以向本项目 反馈。
从PyPI上安装
你可以PyPI上使用以下命令直接安装Colossal-AI。
pip install colossalai
注：现在只支持Linux。
如果你想同时安装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
CUDA_EXT=1 pip install .
如果您不想安装和启用 CUDA 内核融合（使用融合优化器时强制安装），您可以不添加CUDA_EXT=1：
pip install .