github/ColossalAI
1
0
Fork 0
mirror of https://github.com/hpcaitech/ColossalAI.git synced 2025-09-25 11:44:03 +00:00
Code Issues Packages Projects Releases Wiki Activity
Files
feature/ray-rlhf
ColossalAI/applications/ColossalChat/coati/distributed/README.md
Hongxin Liu 43c9b5fb44 [chat] add distributed impl (#6210)
2025-02-21 15:24:23 +08:00

118 B
Raw Permalink Blame History

Requirements

pip install cupy-cuda12x
python -m cupyx.tools.install_library --cuda 12.x --library nccl