Skip to content

Dockerize #5

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Open
dte wants to merge 2 commits into
base: main
Choose a base branch
from
Open

Dockerize #5

Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
2 commits
Select commit Hold shift + click to select a range
a2a7c5a
Create Dockerfile
dte Sep 9, 2021
d518d7b
Create requirements.txt
dte Sep 9, 2021
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
44 changes: 44 additions & 0 deletions Dockerfile
View file
Open in desktop
Original file line number Diff line number Diff line change
@@ -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"]
17 changes: 17 additions & 0 deletions requirements.txt
View file
Open in desktop
Original file line number Diff line number Diff line change
@@ -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