forked from jonkrohn/DLTFpT
-
Notifications
You must be signed in to change notification settings - Fork 0
/
Dockerfile
29 lines (21 loc) · 829 Bytes
/
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
28
29
# Details of the base image are here: hub.docker.com/r/jupyter/scipy-notebook
# Tag [2ce7c06a61a1] is latest image as of August 12, 2019
# It runs Python 3.7.3
FROM jupyter/scipy-notebook:2ce7c06a61a1
MAINTAINER Jon Krohn <[email protected]>
USER $NB_USER
# Install TensorFlow:
RUN pip install tensorflow==2.0.0
# Install PyTorch libraries:
RUN pip install https://download.pytorch.org/whl/cpu/torch-1.0.1.post2-cp37-cp37m-linux_x86_64.whl
RUN pip install torchvision==0.4.0
RUN pip install torchsummary==1.5.1
# Install NLP libraries:
RUN pip install nltk==3.4.5
RUN pip install gensim==3.8.1
RUN pip install spacy==2.2.1
RUN python -m spacy download en_core_web_sm
# Install Reinforcement Learning library:
RUN pip install gym==0.12.5
# Install for Object Detection notebook:
RUN pip install opencv-python==4.1.2.30