diff --git a/katib/hp-tuning.yaml b/katib/hp-tuning.yaml index 58ed22178bdb28f22cc640a8a8fd47a49d6689ed..8202105b6c5cc7c1c8f56a0d557f6c84d860a27a 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