简介
TI-ONE 内置了主流的 TensorFlow、Pytorch、PySpark 等深度学习和机器学习框架,同时推出了自研训练加速框架 Angel,其中 tilearn-llm 是为大模型定制的训练加速组件,已内置在平台通用训练镜像中,同时支持 Notebook 和任务式建模。
内置镜像列表
镜像名称 | 版本 | 支持的训练模式 |
TensorFlow | ti-acc1.0-tf1.15-py3.6-cuda10.0-gpu | PS-Worker |
| tf1.15-py3.7-cpu | PS-Worker,MPI,Horovod |
| tf1.15-py3.7-cuda10.0-gpu | |
| tf2.4-py3.8-cpu | |
| tf2.4-py3.8-cuda11.1-gpu | |
PyTorch | tilearn-llm0.4.2-torch2.1-deepspeed0.10.0-py3.10-cuda12.1-gpu | DDP |
| ti-acc1.0-torch1.9-py3.8-cuda11.1-gpu | DDP |
| torch1.9-py3.8-cuda11.1-gpu | DDP,MPI,Horovod |
Spark | spark2.4.5-cpu | Spark |
PySpark | spark2.4.5-py3.6-cpu | Spark |