- 文档首页
GPU云服务器
GPU云服务器
最佳实践
性能测试/调优
通过单机/多机GPT训练判断GPU性能(Easy_GPT)
性能测试/调优
通过单机/多机GPT训练判断GPU性能(Easy_GPT)
通过单机/多机GPT训练判断GPU性能(Easy_GPT)
本文主要介绍如何在Docker环境中进行端到端单机/多机GPT训练来判断GPU性能是否达标。
背景信息
在处理GPU性能问题时,如果执行了HostDiagnose、Easy_NCCL等操作后仍无法定位到异常或者需要模拟真实业务场景时,您可以通过端到端单机/多机GPT训练,模拟用户真实业务来判断GPU性能是否达标,从而节省依赖安装、编译、配置端口、免密等繁琐步骤。 GPT(Generative Pre-trained Transformer )指OpenAI发布的一系列大型语言模型,它们在大型文本数据集上训练,可被用于文本生成,翻译,分类等任务。GPT系列包括GPT-3,GPT-4等。
机型限制
- 单机训练:支持GPU计算型和高性能计算GPU型实例。
- 多机训练:仅支持高性能计算GPU型实例,且不包括hpcg1ve和hpcpni3h。其中,hpcpni3h实例正在邀测中,如需使用,请联系客户经理申请。
准备环境
- 您已拥有一台或多台GPU实例,本文分别命名为node1、node2。每台实例的基本配置如下:
- 本文GPU实例的镜像以Ubuntu 20.04为例,您也可以任选其它镜像。
/usr/local/cuda/bin/nvcc -V
-
-
-
操作步骤
- 参考文件传输概述,将以下mlp_easy_gpt.py文件上传至云服务器实例。
- 在实例内部执行以下命令,运行mlp_easy_gpt.py脚本。
请将-i后的IP替换为每台实例主网卡的私网IP地址,不同IP之间用“,”分隔,顺序可以互换。如何查看私网IP地址,请参考查看实例信息。 python3 mlp_easy_gpt.py -i <实例的私网IP地址>
python3 mlp_easy_gpt.py -i <node1的私网IP地址,node2的私网IP地址>
结果判定
最近更新时间:2025.03.07 16:29:12