数工科技DIGIT·CN

10 分钟跑通
第一个推理任务

数工开发者文档覆盖 SDK、API、示例、运维手册,帮助你把国产算力能力快速集成进业务系统。

三行代码 调用国产算力

从注册到跑通第一个推理任务,最快仅需 10 分钟。以下示例基于 Python SDK。

01

注册开发者账号

访问 console.digit.cn 注册,完成企业认证即可获得 500 元免费算力额度。

02

安装 SDK

pip install digit-sdk。SDK 自动检测可用 GPU,未安装驱动时回退到 CPU 模式。

03

调用推理接口

通过 Client.infer() 调用云端模型,或加载本地权重在自有硬件上推理。

04

部署到生产

使用 digit-cli 一键构建 Docker 镜像,支持 K8s Operator 部署与滚动升级。

quickstart.py
from digit import Client

client = Client(api_key="dk_live_xxx")

# 云端推理
result = client.infer(
    model="digit-chat-13b",
    prompt="用三句话介绍数工科技",
    max_tokens=256,
)
print(result.text)

# 本地 GPU 推理
from digit.runtime import load_model

model = load_model(
    "/models/digit-chat-13b",
    device="dg:0",   # DG-X900 #0
    dtype="bf16",
)
for token in model.stream("你好"):
    print(token, end="", flush=True)

需要企业级技术支持?

签约客户可获得专属架构师对接、SLA 承诺、源代码级协助。