Skip to content
Snippets Groups Projects
Commit 7aefb38e authored by Dejan Golubovic's avatar Dejan Golubovic
Browse files

Add nvidia-driver mount

parent 1c7ca5e7
No related branches found
No related tags found
No related merge requests found
Hide whitespace changes
Inline Side-by-side
Showing
with 12 additions and 0 deletions
katib/hp-tuning.yaml
+ 12
− 0
View file @ 7aefb38e
......@@ -59,6 +59,10 @@ spec:
- name: krb-secret-vol
secret:
secretName: krb-secret
- name: nvidia-driver
hostPath:
path: /opt/nvidia-driver
type: ""
containers:
- name: pytorch
resources:
......@@ -69,6 +73,8 @@ spec:
mountPath: /eos
- name: krb-secret-vol
mountPath: /secret/krb-secret-vol
- name: nvidia-driver
mountPath: /opt/nvidia-driver
image: gitlab-registry.cern.ch/dholmber/particlenet-images/torchjob:latest
command: [sh, -c]
args:
......@@ -105,6 +111,10 @@ spec:
- name: krb-secret-vol
secret:
secretName: krb-secret
- name: nvidia-driver
hostPath:
path: /opt/nvidia-driver
type: ""
containers:
- name: pytorch
resources:
......@@ -115,6 +125,8 @@ spec:
mountPath: /eos
- name: krb-secret-vol
mountPath: /secret/krb-secret-vol
- name: nvidia-driver
mountPath: /opt/nvidia-driver
image: registry.hub.docker.com/deinal/particlenet-torchjob:multinode-training
command: [sh, -c]
args:
......
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment