modify entrypoint

sillsdev · Jul 5, 2023 · e02b37c · e02b37c
1 parent 413a790
commit e02b37c
Show file tree

Hide file tree

Showing 3 changed files with 4 additions and 3 deletions.
diff --git a/.devcontainer/Dockerfile b/.devcontainer/Dockerfile
@@ -46,4 +46,4 @@ RUN curl "https://awscli.amazonaws.com/awscli-exe-linux-x86_64.zip" -o "awscliv2
     ./aws/install && \
     rm awscliv2.zip
 RUN rm -rf /var/lib/apt/lists/*
-ENTRYPOINT [ "/bin/bash", "-it" ]
+ENTRYPOINT [ "/bin/bash", "-l", "-c" ]
diff --git a/Dockerfile b/Dockerfile
@@ -46,4 +46,4 @@ RUN curl "https://awscli.amazonaws.com/awscli-exe-linux-x86_64.zip" -o "awscliv2
     ./aws/install && \
     rm awscliv2.zip
 RUN rm -rf /var/lib/apt/lists/*
-ENTRYPOINT [ "/bin/bash", "-it" ]
+ENTRYPOINT [ "/bin/bash", "-l", "-c" ]
diff --git a/silnlp/nmt/clearml_connection.py b/silnlp/nmt/clearml_connection.py
@@ -48,7 +48,8 @@ def __post_init__(self) -> None:
             self._load_config()
 
             self.task.set_base_docker(
-                docker_image="nvidia/cuda:11.2.2-cudnn8-runtime-ubuntu20.04"
+                docker_image="ghcr.io/sillsdev/silnlp:latest",
+                docker_arguments="-v /home/clearml/.clearml/hf-cache:/root/.cache/huggingface"
             )
             if self.queue_name.lower() not in ("local", "locally"):
                 self.task.execute_remotely(queue_name=self.queue_name)