Updated base image again to only have transformers change between the images

This commit is contained in:
ebolam 2022-01-26 15:35:36 -05:00
parent d2b15e2a6e
commit a27c441cdf
3 changed files with 12 additions and 6 deletions

View File

@ -5,6 +5,6 @@
FROM mambaorg/micromamba
WORKDIR /content/
COPY environments /content/environments
RUN micromamba install -y -n base -f /content/environments/huggingface.yml
RUN micromamba install -y -n base -f /content/environments/base.yml
USER root
RUN apt update && apt install xorg -y

View File

@ -1,9 +1,8 @@
# 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 environments /content/environments
RUN micromamba install -y -n base -f /content/environments/finetuneanon.yml
USER root
RUN apt update && apt install xorg -y
RUN apt-get update
RUN apt-get install git -y
RUN /opt/conda/bin/pip install tensorflow-gpu git+https://github.com/finetuneanon/transformers.git

View File

@ -0,0 +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 ebolam/koboldai_base
WORKDIR /content/
RUN /opt/conda/bin/pip install transformers