有奖捉虫:办公协同&微信生态&物联网文档专题 HOT

简介

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