Chapter 3. AMD GPU Operator

Next steps

1. Install the Node Feature Discovery Operator.
2. Install the Kernel Module Management Operator.
3. Install and configure the AMD GPU Operator.

Procedure

1. Create a YAML file that tests ROCmInfo:

   Copy to Clipboard Toggle word wrap

   $ 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. Create the rocminfo pod:

   Copy to Clipboard Toggle word wrap

   $ oc create -f rocminfo.yaml

   Example output

   Copy to Clipboard Toggle word wrap

   apiVersion: v1
   pod/rocminfo created

3. Check the rocmnfo log with one MI210 GPU:

   Copy to Clipboard Toggle word wrap

   $ oc logs rocminfo | grep -A5 "Agent"

   Example output

   Copy to Clipboard Toggle word wrap

   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. Delete the pod:

   Copy to Clipboard Toggle word wrap

   $ oc delete -f rocminfo.yaml

   Example output

   Copy to Clipboard Toggle word wrap

   pod "rocminfo" deleted