diff --git a/Dockerfile b/Dockerfile index 35bdec2..abfdf6e 100644 --- a/Dockerfile +++ b/Dockerfile @@ -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 +LABEL maintainer="Jon Krohn " 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