Skip to content

Instantly share code, notes, and snippets.

@rsimd
Created January 7, 2020 07:48
Show Gist options
  • Save rsimd/d6b4e27d41476ea11f6b930d13af1e78 to your computer and use it in GitHub Desktop.
Save rsimd/d6b4e27d41476ea11f6b930d13af1e78 to your computer and use it in GitHub Desktop.
FROM tensorflow/tensorflow:2.0.0-gpu-py3
WORKDIR /libtopicmodel
COPY . /libtopicmodel/
#RUN pip3 install poetry==1.0.0b8
#RUN poetry install
FROM nvidia/cuda:10.1-cudnn7-devel-ubuntu18.04
ARG PYTHON_VERSION=3.7
RUN apt-get update && apt-get install -y --no-install-recommends \
build-essential \
cmake \
git \
curl \
ca-certificates \
libjpeg-dev \
libpng-dev && \
rm -rf /var/lib/apt/lists/*
RUN curl -o ~/miniconda.sh -O https://repo.continuum.io/miniconda/Miniconda3-latest-Linux-x86_64.sh && \
chmod +x ~/miniconda.sh && \
~/miniconda.sh -b -p /opt/conda && \
rm ~/miniconda.sh && \
/opt/conda/bin/conda install -y python=$PYTHON_VERSION \
numpy pyyaml scipy ipython mkl mkl-include ninja cython \
typing tqdm matplotlib seaborn pandas more-itertools click \
gensim pytest jupyter jupyterlab notebook \
pylint flake8 autopep8 tensorflow-gpu && \
/opt/conda/bin/conda clean -ya
/opt/conda/bin/pip install funcy optuna wordcloud pyLDAvis
ENV PATH /opt/conda/bin:$PATH
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment