From 794ec63bbd3cf4f31dd6aeca2334add3e139ec43 Mon Sep 17 00:00:00 2001 From: Richard Chen Date: Tue, 7 May 2019 14:30:00 -0700 Subject: [PATCH] Output the nvidia gpu information in the E2E test. Including the gpu information simplifies driver version verification. nvidia-smi is used in order to display gpu information, which contains the driver version. --- test/e2e/scheduling/nvidia-gpus.go | 9 +++++++++ test/images/cuda-vector-add/Dockerfile | 2 +- 2 files changed, 10 insertions(+), 1 deletion(-) diff --git a/test/e2e/scheduling/nvidia-gpus.go b/test/e2e/scheduling/nvidia-gpus.go index f1fe190d722..ea496027c19 100644 --- a/test/e2e/scheduling/nvidia-gpus.go +++ b/test/e2e/scheduling/nvidia-gpus.go @@ -179,6 +179,7 @@ func testNvidiaGPUs(f *framework.Framework) { // Wait for all pods to succeed for _, pod := range podList { f.PodClient().WaitForSuccess(pod.Name, 5*time.Minute) + logContainers(f, pod) } e2elog.Logf("Stopping ResourceUsageGather") @@ -189,6 +190,14 @@ func testNvidiaGPUs(f *framework.Framework) { framework.ExpectNoError(err, "getting resource usage summary") } +func logContainers(f *framework.Framework, pod *v1.Pod) { + for _, container := range pod.Spec.Containers { + logs, err := e2epod.GetPodLogs(f.ClientSet, f.Namespace.Name, pod.Name, container.Name) + framework.ExpectNoError(err, "Should be able to get container logs for container: %s", container.Name) + e2elog.Logf("Got container logs for %s:\n%v", container.Name, logs) + } +} + var _ = SIGDescribe("[Feature:GPUDevicePlugin]", func() { f := framework.NewDefaultFramework("device-plugin-gpus") ginkgo.It("run Nvidia GPU Device Plugin tests", func() { diff --git a/test/images/cuda-vector-add/Dockerfile b/test/images/cuda-vector-add/Dockerfile index c7b9cfdc31e..968626bf424 100644 --- a/test/images/cuda-vector-add/Dockerfile +++ b/test/images/cuda-vector-add/Dockerfile @@ -23,4 +23,4 @@ RUN apt-get update && apt-get install -y --no-install-recommends \ WORKDIR /usr/local/cuda/samples/0_Simple/vectorAdd RUN make -CMD ./vectorAdd +CMD nvidia-smi && ./vectorAdd