-
Notifications
You must be signed in to change notification settings - Fork 357
/
Dockerfile
27 lines (23 loc) · 1.07 KB
/
Dockerfile
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
# Get the details on the base image here: http://jupyter-docker-stacks.readthedocs.io/en/latest/using/selecting.html#jupyter-scipy-notebook
# Ubuntu 18.04 (bionic) from 2018-04-26
# Note: tag [e7000ca1416d] is latest image as of 06/09/2018
# 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
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