Skip to content

Instantly share code, notes, and snippets.

@Hunsin

Hunsin/jupytorch-cuda.Dockerfile

Last active May 14, 2019 06:33
  • Star 0 You must be signed in to star a gist
  • Fork 0 You must be signed in to fork a gist
Star You must be signed in to star a gist
Embed
What would you like to do?
Learn more about clone URLs
Download ZIP
A Jupyter Dockerfile with PyTorch installed
Raw
jupytorch-cuda.Dockerfile
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
Raw
jupytorch.Dockerfile
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