kubenetes中容器调用GPU

最新推荐文章于 2026-03-27 02:57:59 发布

原创最新推荐文章于 2026-03-27 02:57:59 发布 · 628 阅读

2 ·

本内容遵循CC 4.0 BY-SA版权协议

标签

#kubenetes #GPU

收录于

k8s 专栏收录该内容

2 篇文章

订阅专栏

本文详细介绍如何在Kubernetes环境中配置GPU资源，通过修改yaml文件实现容器任务对GPU的调用。具体步骤包括设置GPU资源限制、挂载数据卷、配置安全上下文等，为深度学习等高性能计算任务提供强大算力支持。

kubenetes中容器调用GPU

在yaml文件中添加GPU资源

apiVersion: batch/v1
kind: Job
metadata:
  name: MyJob
  namespace: myJob
spec:
  template:
    spec:
      containers:
        - name: torch
          image: torch:latest
          resources:
            limits:
              nvidia.com/gpu: 1
          imagePullPolicy: IfNotPresent
          volumeMounts:
            - name: source-data-storage
              mountPath: /input
            - name: result-data-storage
              mountPath: /result
          securityContext:
            privileged: true
      imagePullSecrets:
        - name: regcred-secret
      volumes:
        - name: source-data-storage
          hostPath:
            path: /home/data
        - name: result-data-storage
          hostPath:
            path: /home/result
      restartPolicy: Never