3.3. AMD GPU Operator のテスト

手順

1. ROCmInfo をテストする YAML ファイルを作成します。

   $ cat << EOF > rocminfo.yaml
   
   apiVersion: v1
   kind: Pod
   metadata:
    name: rocminfo
   spec:
    containers:
    - image: docker.io/rocm/pytorch:latest
      name: rocminfo
      command: ["/bin/sh","-c"]
      args: ["rocminfo"]
      resources:
       limits:
         amd.com/gpu: 1
       requests:
         amd.com/gpu: 1
    restartPolicy: Never
   EOF

2. rocminfo Pod を作成します。

   $ oc create -f rocminfo.yaml

   出力例

   apiVersion: v1
   pod/rocminfo created

3. 1 つの MI210 GPU を含む rocmnfo ログを確認します。

   $ oc logs rocminfo | grep -A5 "Agent"

   出力例

   HSA Agents
   ==========
   *******
   Agent 1
   *******
     Name:                    Intel(R) Xeon(R) Gold 6330 CPU @ 2.00GHz
     Uuid:                    CPU-XX
     Marketing Name:          Intel(R) Xeon(R) Gold 6330 CPU @ 2.00GHz
     Vendor Name:             CPU
   --
   Agent 2
   *******
     Name:                    Intel(R) Xeon(R) Gold 6330 CPU @ 2.00GHz
     Uuid:                    CPU-XX
     Marketing Name:          Intel(R) Xeon(R) Gold 6330 CPU @ 2.00GHz
     Vendor Name:             CPU
   --
   Agent 3
   *******
     Name:                    gfx90a
     Uuid:                    GPU-024b776f768a638b
     Marketing Name:          AMD Instinct MI210
     Vendor Name:             AMD

4. Pod を削除します。

   $ oc delete -f rocminfo.yaml

   出力例

   pod "rocminfo" deleted