Skip to content

Instantly share code, notes, and snippets.

@hoangnt2601
Created November 18, 2020 01:35
Show Gist options
  • Select an option

  • Save hoangnt2601/da79d1ec482b729b69ffcdffb3d3e732 to your computer and use it in GitHub Desktop.

Select an option

Save hoangnt2601/da79d1ec482b729b69ffcdffb3d3e732 to your computer and use it in GitHub Desktop.
Build docker image for deep learning env
FROM okwrtdsh/anaconda3:10.0-cudnn7
RUN apt-get update -qq \
&& apt-get install --no-install-recommends -y \
graphviz \
&& apt-get clean \
&& rm -rf /var/lib/apt/lists/*
# opencv install
RUN conda install -c anaconda opencv
# pytorch install
# RUN conda install -c pytorch magma-cuda101
RUN pip install torch==1.7.0+cu101 torchvision==0.8.1+cu101 torchaudio==0.7.0 -f https://download.pytorch.org/whl/torch_stable.html
# tensorflow install
# RUN conda install -c anaconda tensorflow
RUN pip install wrapt --upgrade --ignore-installed
RUN pip install tensorflow
# mxnet install
RUN pip --no-cache-dir install \
mxnet-cu101 \
mxnet-cu101mkl \
graphviz \
pydot_ng \
msgpack
RUN echo "completed!"
CMD ["python3"]
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment