From a2a7c5a7301f579e750b47cec355c704188f4f03 Mon Sep 17 00:00:00 2001 From: Dillon Erb <585865+dte@users.noreply.github.com> Date: Thu, 9 Sep 2021 14:24:43 -0400 Subject: [PATCH 1/2] Create Dockerfile --- Dockerfile | 44 ++++++++++++++++++++++++++++++++++++++++++++ 1 file changed, 44 insertions(+) create mode 100644 Dockerfile diff --git a/Dockerfile b/Dockerfile new file mode 100644 index 0000000..7e5491f --- /dev/null +++ b/Dockerfile @@ -0,0 +1,44 @@ +FROM nvidia/cuda:10.2-cudnn8-devel + +ENV APP_HOME / +WORKDIR $APP_HOME + +RUN apt-get update +RUN apt-get install --yes git curl build-essential wget + +RUN wget https://repo.anaconda.com/miniconda/Miniconda3-latest-Linux-x86_64.sh \ + && mkdir /root/.conda \ + && bash Miniconda3-latest-Linux-x86_64.sh -b + +ENV PATH=$PATH:/root/miniconda3/bin + +RUN conda install -y cudatoolkit=10.2 -c nvidia +RUN conda install -y pytorch torchvision torchaudio -c pytorch -c nvidia +RUN conda install -y pytorch-lightning -c conda-forge + +COPY requirements.txt ./ +RUN pip install --no-cache-dir -r ./requirements.txt + +RUN git clone https://github.com/openai/CLIP +RUN git clone https://github.com/CompVis/taming-transformers.git +RUN git clone https://github.com/BachiLi/diffvg + +# Compile diffvg +WORKDIR diffvg +RUN git submodule update --init --recursive +RUN conda install -y numpy +RUN conda install -y scikit-image +RUN conda install -y -c anaconda cmake +RUN conda install -y -c conda-forge ffmpeg +RUN export DIFFVG_CUDA=1; python setup.py install +WORKDIR .. + + +# Clone clipit last, since it's the second most-likely to change +RUN git clone https://github.com/dribnet/clipit + +RUN conda install -y -c conda-forge jupyterlab + +# Copy local code to container image +# COPY *.py ./ +# CMD ["python3", "pixeldraw.py"] From d518d7b1ec775591ef7b5e5343e5b72b2a6c0b60 Mon Sep 17 00:00:00 2001 From: Dillon Erb <585865+dte@users.noreply.github.com> Date: Thu, 9 Sep 2021 14:25:20 -0400 Subject: [PATCH 2/2] Create requirements.txt --- requirements.txt | 17 +++++++++++++++++ 1 file changed, 17 insertions(+) create mode 100644 requirements.txt diff --git a/requirements.txt b/requirements.txt new file mode 100644 index 0000000..92da6a0 --- /dev/null +++ b/requirements.txt @@ -0,0 +1,17 @@ +ftfy +regex +tqdm +omegaconf +kornia +imageio-ffmpeg +einops +torch-optimizer +easydict +braceexpand +git+https://github.com/pvigier/perlin-numpy +svgwrite +svgpathtools +cssutils +numba +torch-tools +visdom