Skip to content

Instantly share code, notes, and snippets.

Embed
What would you like to do?
A Jupyter Dockerfile with PyTorch installed
FROM continuumio/anaconda3:2019.03
RUN /opt/conda/bin/conda install pytorch torchvision cudatoolkit=10.0 -y --quiet -c pytorch
RUN /opt/conda/bin/conda install jupyter -y --quiet
RUN mkdir /opt/notebooks
EXPOSE 80
CMD jupyter notebook --notebook-dir=/opt/notebooks --ip='0.0.0.0' --port=80 --no-browser --allow-root
FROM continuumio/anaconda3:2019.03
RUN /opt/conda/bin/conda install pytorch-cpu torchvision-cpu -y --quiet -c pytorch
RUN /opt/conda/bin/conda install jupyter -y --quiet
RUN mkdir /opt/notebooks
EXPOSE 80
CMD jupyter notebook --notebook-dir=/opt/notebooks --ip='0.0.0.0' --port=80 --no-browser --allow-root
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment