fix: added script for installation of software

Signed-off-by: Akash Jaiswal <akashjaiswal3846@gmail.com>

Author: jaiakash

.github/workflows/test-e2e-gpu.yaml

@@ -44,6 +44,11 @@ jobs:
           pip install papermill==2.6.0 jupyter==1.1.1 ipykernel==6.29.5
           pip install git+https://github.com/kubeflow/sdk.git@main
 
+      - name: Setup the ephermal VM with software
+        working-directory: ${{ env.GOPATH }}/src/github.com/kubeflow/trainer
+        run: |
+          make test-e2e-gpu-install-software
+
       - name: Setup cluster with GPU support using nvidia/kind
         working-directory: ${{ env.GOPATH }}/src/github.com/kubeflow/trainer
         run: |

Makefile

@@ -202,6 +202,10 @@ test-e2e-setup-cluster: kind ## Setup Kind cluster for e2e test.
 test-e2e-setup-gpu-cluster: kind ## Setup Kind cluster for GPU e2e test.
 	KIND=$(KIND) K8S_VERSION=$(K8S_VERSION) ./hack/e2e-setup-gpu-cluster.sh
 
+.PHONY: test-e2e-gpu-install-software
+test-e2e-gpu-install-software: ## Install software for GPU e2e test.
+	./hack/e2e-gpu-arc-software.sh
+
 .PHONY: test-e2e
 test-e2e: ginkgo ## Run Go e2e test.
 	$(GINKGO) -v ./test/e2e/...

hack/e2e-gpu-install-software.sh

@@ -0,0 +1,31 @@
+# Installing nvidia toolkit
+echo "Install nvidia-ctk tool"
+
+curl -fsSL https://nvidia.github.io/libnvidia-container/gpgkey | sudo gpg --dearmor -o /usr/share/keyrings/nvidia-container-toolkit-keyring.gpg \
+  && curl -s -L https://nvidia.github.io/libnvidia-container/stable/deb/nvidia-container-toolkit.list | \
+    sed 's#deb https://#deb [signed-by=/usr/share/keyrings/nvidia-container-toolkit-keyring.gpg] https://#g' | \
+    sudo tee /etc/apt/sources.list.d/nvidia-container-toolkit.list
+
+sudo apt-get update
+
+export NVIDIA_CONTAINER_TOOLKIT_VERSION=1.17.8-1
+sudo apt-get install -y \
+    nvidia-container-toolkit=${NVIDIA_CONTAINER_TOOLKIT_VERSION} \
+    nvidia-container-toolkit-base=${NVIDIA_CONTAINER_TOOLKIT_VERSION} \
+    libnvidia-container-tools=${NVIDIA_CONTAINER_TOOLKIT_VERSION} \
+    libnvidia-container1=${NVIDIA_CONTAINER_TOOLKIT_VERSION}
+
+# Install nvkind
+echo "Install nvkind"
+sudo go install github.com/NVIDIA/nvkind/cmd/nvkind@latest
+
+# Few Testing commands
+go version
+which nvkind
+
+# Alias kubectl, kind, helm, nvkind with sudo
+alias docker="sudo docker"
+alias kubectl="sudo kubectl"
+alias kind="sudo kind"
+alias helm="sudo helm"
+alias nvkind="sudo nvkind"