machine_learning_image/Dockerfile at master · cmudeeplearning11785/machine_learning_image · GitHub

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
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
# https://hub.docker.com/r/kaixhin/cuda-theano/~/dockerfile/
# https://github.com/autolab/Tango/blob/master/vmms/Dockerfile
# https://github.com/tensorflow/tensorflow/blob/master/tensorflow/tools/docker/Dockerfile
# https://github.com/pytorch/pytorch/blob/master/Dockerfile
# Autolab - autograding docker image for machine learning

FROM cmudeeplearning11785/autograding_image
MAINTAINER Benjamin Striner <bstriner@gmail.com>

# Python3
RUN apt-get update --fix-missing
RUN apt-get install -y gcc
RUN apt-get install -y make
RUN apt-get install -y build-essential
RUN apt-get update && apt-get install -y \
  # gfortran \
  git \
  # wget \
  # curl \
  # cmake \
  pkg-config \
  python3 \
  # python3-dev \
  # liblapack-dev \
  # libopenblas-dev \
  rsync \
  software-properties-common \
  unzip \
  # libfreetype6-dev \
  # libpng12-dev \
  # libzmq3-dev \
  python3-pip \
  python3-nose \
  python3-numpy \
  python3-scipy
RUN pip3 install --upgrade pip
RUN pip3 install --upgrade six
RUN pip3 install pyOpenSSL ndg-httpsclient pyasn1

# cmake
# WORKDIR /tmp
# RUN wget http://www.cmake.org/files/v3.10/cmake-3.10.0.tar.gz
# RUN tar xf cmake-3.10.0.tar.gz
# RUN cd cmake-3.10.0 && ./configure && make && make install

# Tensorflow
RUN pip3 install \
        Pillow \
        h5py \
        ipykernel \
        jupyter \
        matplotlib \
		scipy \
		sympy \
        pandas \
        sklearn \
        && \
    python3 -m ipykernel.kernelspec
RUN pip3 install tensorflow

# Pytorch
# WORKDIR /tmp
# RUN pip3 install pyyaml mkl setuptools cmake cffi
# RUN git clone --recursive https://github.com/pytorch/pytorch.git && cd pytorch && python3 setup.py install
# RUN pip3 install --no-deps torchvision
# RUN rm -Rf pytorch
# RUN git clone https://github.com/pytorch/vision.git && cd vision && pip install -v .
RUN pip3 install http://download.pytorch.org/whl/cu80/torch-0.3.0.post4-cp35-cp35m-linux_x86_64.whl
RUN pip3 install torchvision
RUN pip3 install inferno-pytorch
RUN pip3 install git+https://github.com/pytorch/tnt.git@master

# Misc Python
RUN pip3 install pytest nose nose-parameterized pytest-pep8 pytest-helpers-namespace
RUN pip3 install h5py lmdb tqdm

# Datasets
# RUN mkdir /data
# WORKDIR /data
# RUN wget https://www.cs.toronto.edu/~kriz/cifar-10-python.tar.gz
# RUN tar xzf cifar-10-python.tar.gz
# RUN rm cifar-10-python.tar.gz
# ENV DATA_PATH=/data
# ENV CIFAR10_PATH=/data/cifar-10-batches-py

# Cleanup
WORKDIR /home
RUN apt-get remove -y git
# RUN apt-get remove -y wget curl cmake gfortran build-essential gcc
RUN apt-get clean
RUN apt-get -y autoremove

# Check installation
RUN ls -l /home
RUN which autodriver