Skip to content
Open
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
30 changes: 15 additions & 15 deletions Dockerfile
Original file line number Diff line number Diff line change
Expand Up @@ -4,24 +4,24 @@
# I've specified exact versions here. Users can modify at their own risk to get more recent versions of libraries.
# Note that this base image includes Python 3.6.5 as part of Miniconda 4.5.1

FROM jupyter/scipy-notebook:e7000ca1416d
FROM jupyter/scipy-notebook:latest

MAINTAINER Jon Krohn <[email protected]>
LABEL maintainer="Jon Krohn <[email protected]>"

USER $NB_USER

# install the libraries
RUN \
# Upgrade pip first
pip install --upgrade pip==19.0.3 && \
# install TensorFlow library
pip install --upgrade https://storage.googleapis.com/tensorflow/linux/cpu/tensorflow-1.8.0-cp36-cp36m-linux_x86_64.whl && \
# install TFlearn, Keras, PyTorch, Torchvision, NLP libraries and RL environment
pip install tflearn==0.3.2 \
keras==2.2.0 \
nltk==3.3 \
gensim==3.4.0 \
gym==0.9.6 \
https://download.pytorch.org/whl/cpu/torch-1.0.1.post2-cp36-cp36m-linux_x86_64.whl \
torchvision==0.2.2.post3 \
h5py==2.8.0
# Upgrade pip first
pip install --upgrade pip && \
# install TensorFlow library
pip install tensorflow==2.17.0 && \
# install TFlearn, Keras, PyTorch, Torchvision, NLP libraries and RL environment
pip install tflearn==0.5.0 \
keras==3.6.0 \
nltk==3.9.1 \
gensim==4.3.3 \
gym==0.26.2 \
torch==2.5.0 \
torchvision==0.20.0 \
h5py==3.10.0