Created
November 18, 2020 01:35
-
-
Save hoangnt2601/da79d1ec482b729b69ffcdffb3d3e732 to your computer and use it in GitHub Desktop.
Build docker image for deep learning env
This file contains hidden or bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| 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