diff --git a/tests/gpu-pod.yaml b/tests/gpu-pod.yaml index 055c914d3..97a084f69 100644 --- a/tests/gpu-pod.yaml +++ b/tests/gpu-pod.yaml @@ -3,6 +3,7 @@ kind: Pod metadata: name: gpu-operator-test spec: + runtimeClassName: nvidia restartPolicy: OnFailure containers: - name: cuda-vector-add