在本地开始

你可以在下面选择你的PyTorch以及CUDA版本,并下载对应的Colossal-AI。由于PyTorch并没有为每个CUDA版本发布对应的版本,所以您如果没有找到可用的下载,可能需要从源码安装Colossal-AI。如果想要安装预览版本,您需要Python 3.8以及最新版本的PyTorch和CUDA.


ColossalAI version:
PyTorch version:
CUDA version:
Run this command:
Not available. Please install from source.