Minimal Dockerfile.

Author: hameerabbasi

.dockerignore

@@ -0,0 +1 @@
+.gitignore

docker.init.sh

@@ -0,0 +1,47 @@
+#!/usr/bin/env bash
+
+# Setup ssh
+setup_ssh() {
+    if [[ $PUBLIC_KEY ]]; then
+        echo "Setting up SSH..."
+        mkdir -p ~/.ssh
+        echo "$PUBLIC_KEY" >> ~/.ssh/authorized_keys
+        chmod 700 -R ~/.ssh
+
+        if [ ! -f /etc/ssh/ssh_host_rsa_key ]; then
+            ssh-keygen -t rsa -f /etc/ssh/ssh_host_rsa_key -q -N ''
+            echo "RSA key fingerprint:"
+            ssh-keygen -lf /etc/ssh/ssh_host_rsa_key.pub
+        fi
+
+        if [ ! -f /etc/ssh/ssh_host_dsa_key ]; then
+            ssh-keygen -t dsa -f /etc/ssh/ssh_host_dsa_key -q -N ''
+            echo "DSA key fingerprint:"
+            ssh-keygen -lf /etc/ssh/ssh_host_dsa_key.pub
+        fi
+
+        if [ ! -f /etc/ssh/ssh_host_ecdsa_key ]; then
+            ssh-keygen -t ecdsa -f /etc/ssh/ssh_host_ecdsa_key -q -N ''
+            echo "ECDSA key fingerprint:"
+            ssh-keygen -lf /etc/ssh/ssh_host_ecdsa_key.pub
+        fi
+
+        if [ ! -f /etc/ssh/ssh_host_ed25519_key ]; then
+            ssh-keygen -t ed25519 -f /etc/ssh/ssh_host_ed25519_key -q -N ''
+            echo "ED25519 key fingerprint:"
+            ssh-keygen -lf /etc/ssh/ssh_host_ed25519_key.pub
+        fi
+
+        service ssh start
+
+        echo "SSH host keys:"
+        for key in /etc/ssh/*.pub; do
+            echo "Key: $key"
+            ssh-keygen -lf $key
+        done
+    fi
+}
+
+setup_ssh
+service ssh start
+sleep infinity

resources/docker/Dockerfile

@@ -0,0 +1,29 @@
+#To build, run
+#    docker build -t <image-name> . -f resources/docker/Dockerfile
+#    docker tag <image-name> <dockerhub-username>/<repository-name>:<tag>
+#    docker push <dockerhub-username>/<repository-name>:<tag>
+
+FROM --platform=linux/amd64 ubuntu:22.04
+
+ENV OT_PLATFORM=cuda
+ENV OT_SCRIPT_DEBUG=true
+ENV PATH=/root/.pixi/bin:${PATH}
+ENV TZ=Etc/UTC
+
+WORKDIR /
+RUN apt-get update && \
+    apt-get install -y \
+    git \
+    curl \
+    ssh
+ADD . /OneTrainer
+
+RUN cd OneTrainer \
+    && export OT_PLATFORM=cuda \
+    && export OT_SCRIPT_DEBUG=true \
+    && ./install.sh \
+    && rm -rf /root/.cache \
+    && mkdir -p /workspace \
+    && ln -s /OneTrainer /workspace/OneTrainer
+
+ENTRYPOINT [ "/usr/bin/bash /docker.init.sh" ]