1
DCU工具
提问帖
有回复
新人贴
DCU如何物理独占3张k100中的2张
2026-01-29 14:19:31
140

场景:一个有3张k100服务器,已经部署了hami模式dcu,现在想让k8s集群独占其中两张卡,集群中的业务pod都使用指定的这两张k100,另外一张卡供其余业务使用,具体该怎么操作

源码中的示例:

apiVersion: v1
kind: Pod
metadata:
  name: alexnet-tf-gpu-pod
  labels:
    purpose: demo-tf-amdgpu
spec:
  containers:
    - name: alexnet-tf-gpu-container
      image: rocm/tensorflow:latest
      workingDir: /root
      env:
      - name: HIP_VISIBLE_DEVICES
        value: "0" # # 0,1,2,...,n for running on GPU and select the GPUs, -1 for running on CPU 
      command: ["/bin/bash", "-c", "--"]
      args: ["python3 benchmarks/scripts/tf_cnn_benchmarks/tf_cnn_benchmarks.py --model=alexnet; trap : TERM INT; sleep infinity & wait"]
      resources:
        limits:
          dcu/gpu: 1 # requesting a GPU

env中的HIP_VISIBLE_DEVICES这个参数是否是通用的,能否加入到dcu-device-plugin中,让GPU注册的时候就只注册指定的GPU


全部评论(1)
只看楼主
W
wangzhq
1楼
https://developer.sourcefind.cn/document/87ee5c5b-c10d-11f0-b077-0242ac150003?id=8df80ff9-c10e-11f0-b077-0242ac150003
DCU-Device-Plugin的用法如上,目前不支持指定DCU注册!
2/1
没有更多内容了
--
楼主
Level.1
0 帖子
楼主热帖
暂无数据