王海鱼 发表于 2025-3-28 09:15:41

[AIGC模型训练推理] Open-Clip基于DevServer适配PyTorch NPU训练指导

Open-Clip广泛应用于AIGC和多模态视频编码器的训练。
方案概览

本方案先容了在ModelArts的DevServer上使用昇腾NPU盘算资源开展Open-clip训练的详细过程。完本钱方案的摆设,必要先联系您所在企业的华为方技能支持购买DevServer资源。
本方案现在仅适用于企业客户。
资源规格要求

推荐使用“西南-贵阳一”Region上的DevServer资源和Ascend Snt9B单机单卡。
    表1 情况要求       模型
   版本
           CANN
    cann_8.0.rc1
    PyTorch
    pytorch_2.1.0

获取镜像

    表2 获取镜像       分类
   名称
   获取路径
           基础镜像
    西南-贵阳一:swr.cn-southwest-2.myhuaweicloud.com/atelier/pytorch_2_1_ascend:pytorch_2.1.0-cann_8.0.rc1-py_3.9-hce_2.0.2312-aarch64-snt9b-20240516142953-ca51f42
    从SWR拉取。
获取软件

本教程使用的是Open-clip源码包。
昇腾适配过程通过修改训练脚本方式实现,不涉及其他软件获取。
Step1 准备情况


[*]请参考DevServer资源开通,购买DevServer资源,并确保呆板已开通,密码已获取,能通过SSH登录,不同呆板之间网络互通。   阐明:
[*] 当容器必要提供服务给多个用户,大概多个用户共享使用该容器时,应限定容器访问Openstack的管理地址(169.254.169.254),以防止容器获取宿主机的元数据。具体操纵请参见禁止容器获取宿主机元数据。

免责声明:如果侵犯了您的权益,请联系站长,我们会及时删除侵权内容,谢谢合作!更多信息从访问主页:qidao123.com:ToB企服之家,中国第一个企服评测及商务社交产业平台。
页: [1]
查看完整版本: [AIGC模型训练推理] Open-Clip基于DevServer适配PyTorch NPU训练指导