From d2b15e2a6ed379c1329fa516361e74f4f93a4406 Mon Sep 17 00:00:00 2001 From: ebolam Date: Wed, 26 Jan 2022 11:35:58 -0500 Subject: [PATCH] Updated dockerfiles to create images for docker hub that are per-compiled --- docker-cuda/Dockerfile | 8 ++------ docker-cuda/Dockerfile_base | 10 ++++++++++ docker-cuda/Dockerfile_base_finetune | 9 +++++++++ docker-cuda/Dockerfile_finetune | 8 ++------ 4 files changed, 23 insertions(+), 12 deletions(-) create mode 100644 docker-cuda/Dockerfile_base create mode 100644 docker-cuda/Dockerfile_base_finetune diff --git a/docker-cuda/Dockerfile b/docker-cuda/Dockerfile index e863ea58..36dac8e1 100644 --- a/docker-cuda/Dockerfile +++ b/docker-cuda/Dockerfile @@ -1,11 +1,7 @@ # This dockerfile is meant to serve as a rocm base image. It registers the debian rocm package repository, and # installs the rocm-dev package. -FROM mambaorg/micromamba +FROM ebolam/koboldai_base WORKDIR /content/ COPY . /content/ -RUN ls -RUN micromamba install -y -n base -f /content/environments/huggingface.yml -USER root -RUN apt update && apt install xorg -y -CMD python aiserver.py \ No newline at end of file +CMD python aiserver.py diff --git a/docker-cuda/Dockerfile_base b/docker-cuda/Dockerfile_base new file mode 100644 index 00000000..c026f9df --- /dev/null +++ b/docker-cuda/Dockerfile_base @@ -0,0 +1,10 @@ +# This dockerfile is meant to serve as a rocm base image. It registers the debian rocm package repository, and +# installs the rocm-dev package. + + +FROM mambaorg/micromamba +WORKDIR /content/ +COPY environments /content/environments +RUN micromamba install -y -n base -f /content/environments/huggingface.yml +USER root +RUN apt update && apt install xorg -y diff --git a/docker-cuda/Dockerfile_base_finetune b/docker-cuda/Dockerfile_base_finetune new file mode 100644 index 00000000..4fd7d0ff --- /dev/null +++ b/docker-cuda/Dockerfile_base_finetune @@ -0,0 +1,9 @@ +# This dockerfile is meant to serve as a rocm base image. It registers the debian rocm package repository, and +# installs the rocm-dev package. + +FROM mambaorg/micromamba +WORKDIR /content/ +COPY environments /content/environments +RUN micromamba install -y -n base -f /content/environments/finetuneanon.yml +USER root +RUN apt update && apt install xorg -y diff --git a/docker-cuda/Dockerfile_finetune b/docker-cuda/Dockerfile_finetune index 39e13d98..cd8f4b4b 100644 --- a/docker-cuda/Dockerfile_finetune +++ b/docker-cuda/Dockerfile_finetune @@ -1,11 +1,7 @@ # This dockerfile is meant to serve as a rocm base image. It registers the debian rocm package repository, and # installs the rocm-dev package. -FROM mambaorg/micromamba +FROM ebolam/koboldai_base:finetune WORKDIR /content/ COPY . /content/ -RUN ls -RUN micromamba install -y -n base -f /content/environments/finetuneanon.yml -USER root -RUN apt update && apt install xorg -y -CMD python aiserver.py \ No newline at end of file +CMD python aiserver.py