Skip to content

Instantly share code, notes, and snippets.

@Hunsin

Hunsin/jupytorch-cuda.Dockerfile

Last active May 14, 2019 06:33
Show Gist options
  • Save Hunsin/d0af83b2c114a79f25746283142c61c5 to your computer and use it in GitHub Desktop.
Save Hunsin/d0af83b2c114a79f25746283142c61c5 to your computer and use it in GitHub Desktop.
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