-
Notifications
You must be signed in to change notification settings - Fork 11
/
Dockerfile
executable file
·106 lines (83 loc) · 4.25 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
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
96
97
98
99
100
101
102
103
104
105
106
FROM nvidia/cuda:8.0-cudnn7-devel-ubuntu16.04
# FROM ubuntu:16.04
MAINTAINER ANONYMOUS
# CUDA includes
ENV CUDA_PATH /usr/local/cuda
ENV CUDA_INCLUDE_PATH /usr/local/cuda/include
ENV CUDA_LIBRARY_PATH /usr/local/cuda/lib64
# Ubuntu Packages
RUN apt-get update -y && apt-get install software-properties-common -y && \
add-apt-repository -y multiverse && apt-get update -y && apt-get upgrade -y && \
apt-get install -y apt-utils nano vim man build-essential wget sudo && \
rm -rf /var/lib/apt/lists/*
# Install curl and other dependencies
RUN apt-get update -y && apt-get install -y curl libssl-dev openssl libopenblas-dev \
libhdf5-dev hdf5-helpers hdf5-tools libhdf5-serial-dev libprotobuf-dev protobuf-compiler git
RUN curl -sk https://raw.githubusercontent.com/torch/distro/master/install-deps | bash && \
rm -rf /var/lib/apt/lists/*
#Install python3 pip3
RUN apt-get update
RUN apt-get update && apt-get -y install python3 python3-pip
RUN pip3 install --upgrade pip
RUN pip3 install numpy scipy pyyaml matplotlib
RUN pip3 install imageio
RUN pip3 install tensorboard-logger
RUN mkdir /install
WORKDIR /install
#### -------------------------------------------------------------------
#### install MongoDB (for Sacred)
#### -------------------------------------------------------------------
# Install pymongo
RUN pip3 install pymongo
#### -------------------------------------------------------------------
#### install pysc2 #(from Mika fork)
#### -------------------------------------------------------------------
# RUN git clone https://github.com/samvelyan/pysc2.git /install/pysc2
#RUN git clone https://github.com/deepmind/pysc2.git /install/pysc2 && cd /install/pysc2 && git checkout 65f8badf1b3cbc0d711a8d4c87e4501225b1c0fa
#RUN pip3 install /install/pysc2/
#RUN pip3 install /install/pysc2/
RUN git clone https://github.com/deepmind/pysc2.git /install/pysc2
RUN pip3 install /install/pysc2/
#### -------------------------------------------------------------------
#### install Sacred
#### -------------------------------------------------------------------
RUN pip3 install setuptools
RUN git clone https://github.com/idsia/sacred.git /install/sacred && cd /install/sacred && python3 setup.py install
#### -------------------------------------------------------------------
#### final steps
#### -------------------------------------------------------------------
RUN pip3 install pygame
#### -------------------------------------------------------------------
#### Plotting tools
#### -------------------------------------------------------------------
# RUN apt-get -y install ipython ipython-notebook
RUN pip3 install statsmodels pandas seaborn
RUN mkdir /pool && echo "export PATH=$PATH:'/pool/pool'" >> ~/.bashrc
RUN pip3 install cloudpickle ruamel.yaml
RUN apt-get install -y libhdf5-serial-dev cmake
RUN git clone https://github.com/Blosc/c-blosc.git /install/c-blosc && cd /install/c-blosc && cmake -DCMAKE_INSTALL_PREFIX=/usr/local && cmake --build . --target install
RUN pip3 install tables h5py
#### -------------------------------------------------------------------
#### install tensorflow
#### -------------------------------------------------------------------
RUN pip3 install tensorflow-gpu
#RUN pip3 install --upgrade https://storage.googleapis.com/tensorflow/linux/gpu/tensorflow_gpu-1.6.0-cp34-cp34m-linux_x86_64.whl
#### -------------------------------------------------------------------
#### install pytorch
#### -------------------------------------------------------------------
# RUN git clone https://github.com/csarofeen/pytorch /install/pytorch && cd /install/pytorch
# RUN pip3 install numpy pyyaml mkl setuptools cffi
# RUN apt-get install -y cmake gcc
# RUN cd /install/pytorch && python3 setup.py install
#RUN pip3 install http://download.pytorch.org/whl/cu80/torch-0.2.0.post3-cp35-cp35m-manylinux1_x86_64.whl
RUN pip3 install torch
#RUN pip3 install http://download.pytorch.org/whl/cu90/torch-0.3.1-cp35-cp35m-linux_x86_64.whl
RUN pip3 install torchvision snakeviz pytest probscale
RUN apt-get install -y htop iotop
EXPOSE 8888
WORKDIR /fastmarl
# RUN echo "mongod --fork --logpath /var/log/mongod.log" >> ~/.bashrc
#CMD ["mongod", "--fork", "--logpath", "/var/log/mongod.log"]
# EXPOSE 27017
# EXPOSE 28017
# CMD service mongod start && tail -F /dev/null