From 7db070f96c8415d343cd399df9f63e2f1fcee54f Mon Sep 17 00:00:00 2001
From: Dejan Golubovic <dejan.golubovic@cern.ch>
Date: Thu, 24 Mar 2022 12:08:31 +0100
Subject: [PATCH] Add nvidia driver envs

---
 katib/hp-tuning.yaml | 10 ++++++++++
 1 file changed, 10 insertions(+)

diff --git a/katib/hp-tuning.yaml b/katib/hp-tuning.yaml
index 58ed221..8202105 100644
--- a/katib/hp-tuning.yaml
+++ b/katib/hp-tuning.yaml
@@ -68,6 +68,11 @@ spec:
                     resources: 
                       limits:
                         nvidia.com/gpu: 1
+                    env:
+                    - name: PATH
+                      value: /bin:/usr/bin:/usr/local/bin:/opt/nvidia-driver/bin
+                    - name: LD_LIBRARY_PATH
+                      value: /opt/nvidia-driver/lib64
                     volumeMounts:
                       - name: eos
                         mountPath: /eos
@@ -120,6 +125,11 @@ spec:
                     resources: 
                       limits:
                         nvidia.com/gpu: 1
+                    env:
+                    - name: PATH
+                      value: /bin:/usr/bin:/usr/local/bin:/opt/nvidia-driver/bin
+                    - name: LD_LIBRARY_PATH
+                      value: /opt/nvidia-driver/lib64
                     volumeMounts:
                       - name: eos
                         mountPath: /eos
-- 
GitLab