Skip to content

Instantly share code, notes, and snippets.

@stared

stared/Dockerfile

Created January 8, 2018 19:16
Show Gist options
  • Star 0 You must be signed in to star a gist
  • Fork 0 You must be signed in to fork a gist
  • Save stared/123a246bb0ec1a9efb915592a79a1f09 to your computer and use it in GitHub Desktop.
Save stared/123a246bb0ec1a9efb915592a79a1f09 to your computer and use it in GitHub Desktop.
Download ZIP
Docker with TensorFlow and PyTorch
Raw
Dockerfile
FROM jupyter/scipy-notebook
# Install Tensorflow and PyTorch
RUN conda install --quiet --yes \
'tensorflow=1.3*' \
'keras=2.0*' \
'pytorch=0.3' \
'torchvision' && \
conda clean -tipsy && \
fix-permissions $CONDA_DIR
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment