SEVN/Singularity at master · mweiss17/SEVN · 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
#This is a dockerfile that sets up a full SEVN install
Bootstrap: docker

# Here we ll build our container upon the ubuntu container
From: ubuntu:18.04

# Export global environment variables
%environment
        export PYTHONPATH="/usr/local/:$PYTHONPATH"
        export SHELL=/bin/bash
        export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:/usr/local/bin
        export PATH=/gym/.tox/py3/bin:$PATH

# Then we put everything we need to install
%post
        apt -y update && \
        apt -y upgrade && \
        apt install -y \
        python3-setuptools \
        python3-dev \
        python3-opengl \
        python3-pip \
        libjpeg-dev \
        libboost-all-dev \
        libsdl2-dev \
        libosmesa6-dev \
        patchelf \
        ffmpeg \
        xvfb \
        wget \
        git \
        unzip && \
        apt clean && \
        rm -rf /var/lib/apt/lists/*

        # export env vars
        . /environment

        # create default mount points
        echo "Creating mount points"
        mkdir /scratch
        mkdir /data
        mkdir /dataset
        mkdir /tmp_log
        mkdir /final_log


        pip3 install ipdb tensorflow torch==1.2
        cd scratch/

        # Download Gym
        git clone https://github.com/openai/gym.git || true && \
        cd gym
        pip3 install -e .
        cd ..

        # install SEVN and dependencies
        git clone https://github.com/openai/baselines.git
        cd baselines
        pip3 install -e .
        cd ..
        git clone https://github.com/mweiss17/SEVN.git
        cd SEVN
        pip3 install -e .
        cd ..
        git clone https://github.com/simonchamorro/NAVI-STR
        cd NAVI-STR
        pip3 install -e .
        cd ..
        mv NAVI-STR NAVI_STR

        # Install pytorch-a2c-ppo-acktr
        git clone https://github.com/simonchamorro/SEVN-model.git
        cd SEVN-model
        mkdir trained_models/
        mkdir trained_models/ppo/
	pip3 install -e .

%runscript
        exec /bin/bash "$@"