ClearML/clearml-agent
1
0
Fork 0
mirror of https://github.com/clearml/clearml-agent synced 2025-01-31 09:06:52 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

Fix nvidia pytorch dockers support

Browse Source
This commit is contained in:
allegroai 2020-11-25 16:45:09 +02:00
parent 64e198a57a
commit a8c543ef7b
1 changed files with 3 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
trains_agent/glue/k8s.py
View File

@ -49,6 +49,8 @@ class K8sIntegration(Worker):
"sed 's@session\s*required\s*pam_loginuid.so@session optional pam_loginuid.so@g' -i /etc/pam.d/sshd ; " \
"echo 'AcceptEnv TRAINS_API_ACCESS_KEY TRAINS_API_SECRET_KEY' >> /etc/ssh/sshd_config ; " \
'echo "export VISIBLE=now" >> /etc/profile ; ' \
'echo "export PATH=$PATH" >> /etc/profile ; ' \
'echo "ldconfig" >> /etc/profile ; ' \
"/usr/sbin/sshd -p {port} & ; "
CONTAINER_BASH_SCRIPT = \
@ -327,7 +329,7 @@ class K8sIntegration(Worker):
container = merge_dicts(
container,
dict(name=name, image=docker_image,
command=['/bin/sh'],
command=['/bin/bash'],
args=['-c', create_trains_conf + self.container_bash_script.format(
extra_bash_init_cmd=self.extra_bash_init_script,
task_id=task_id)]))