lorax/Dockerfile.dev at 699a5e995abf9fe368f23b30ab77134ab5f3c23f · predibase/lorax · 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
# LoRAX base image
FROM ghcr.io/predibase/lorax:main as base

# Update and install system packages
RUN apt update && \
    DEBIAN_FRONTEND=noninteractive apt install pkg-config rsync tmux vim rust-gdb git -y

# Install protoc
RUN PROTOC_ZIP=protoc-21.12-linux-x86_64.zip && \
    curl -OL https://github.com/protocolbuffers/protobuf/releases/download/v21.12/$PROTOC_ZIP && \
    unzip -o $PROTOC_ZIP -d /usr/local bin/protoc && \
    unzip -o $PROTOC_ZIP -d /usr/local 'include/*' && \
    rm -f $PROTOC_ZIP

# Refresh hash table
RUN hash -r

# Install cmake via pip
RUN pip install cmake

# Clone lorax repo
RUN git clone https://github.com/predibase/lorax.git /lorax


RUN echo "PS1='${debian_chroot:+($debian_chroot)}\[\033[01;32m\]\u@\h\[\033[00m\]:\[\033[01;34m\]\w\[\033[00m\]\$ '" > /root/.bashrc


# Create .tmux.conf with vim-tmux navigation config
RUN echo '# .tmux.conf\n\
    # Smart pane switching with awareness of Vim splits.\n\
    # See: https://github.com/christoomey/vim-tmux-navigator\n\
    is_vim="ps -o state= -o comm= -t '"'"'#{pane_tty}'"'"' \\\n\
    | grep -iqE '"'"'^[^TXZ ]+ +(\\\\S+\\\\/)?g?(view|n?vim?x?)(diff)?$'"'"'"\n\
    bind-key -n '"'"'C-h'"'"' if-shell "$is_vim" '"'"'send-keys C-h'"'"'  '"'"'select-pane -L'"'"'\n\
    bind-key -n '"'"'C-j'"'"' if-shell "$is_vim" '"'"'send-keys C-j'"'"'  '"'"'select-pane -D'"'"'\n\
    bind-key -n '"'"'C-k'"'"' if-shell "$is_vim" '"'"'send-keys C-k'"'"'  '"'"'select-pane -U'"'"'\n\
    bind-key -n '"'"'C-l'"'"' if-shell "$is_vim" '"'"'send-keys C-l'"'"'  '"'"'select-pane -R'"'"'\n\
    \n\
    tmux_version='"'"'$(tmux -V | sed -En "s/^tmux ([0-9]+(.[0-9]+)?).*/\\1/p")'"'"'\n\
    if-shell -b '"'"'[ "$(echo "$tmux_version < 3.0" | bc)" = 1 ]'"'"' \\\n\
    "bind-key -n '"'"'C-\\\\'"'"' if-shell \\"$is_vim\\" '"'"'send-keys C-\\\\'"'"'  '"'"'select-pane -l'"'"'"\n\
    if-shell -b '"'"'[ "$(echo "$tmux_version >= 3.0" | bc)" = 1 ]'"'"' \\\n\
    "bind-key -n '"'"'C-\\\\'"'"' if-shell \\"$is_vim\\" '"'"'send-keys C-\\\\\\\\'"'"'  '"'"'select-pane -l'"'"'"\n\
    \n\
    bind-key -T copy-mode-vi '"'"'C-h'"'"' select-pane -L\n\
    bind-key -T copy-mode-vi '"'"'C-j'"'"' select-pane -D\n\
    bind-key -T copy-mode-vi '"'"'C-k'"'"' select-pane -U\n\
    bind-key -T copy-mode-vi '"'"'C-l'"'"' select-pane -R\n\
    bind-key -T copy-mode-vi '"'"'C-\\\\'"'"' select-pane -l\n\
    \n\
    set-window-option -g automatic-rename on\n\
    set -g default-terminal "screen-256color"\n\
    \n\
    # Mouse options\n\
    set -g mouse on\n\
    bind -n WheelUpPane   select-pane -t= \\; copy-mode -e \\; send-keys -M\n\
    bind -n WheelDownPane select-pane -t= \\;                 send-keys -M\n\
    \n\
    # enable wm window titles\n\
    set -g set-titles on\n\
    set -g set-titles-string "tmux.#I.#W"\n\
    \n\
    # Act like Vim\n\
    setw -g mode-keys vi\n\
    \n\
    # Vim-like pane navigation\n\
    bind C-h select-pane -L\n\
    bind h select-pane -L\n\
    bind C-j select-pane -D\n\
    bind j select-pane -D\n\
    bind C-k select-pane -U\n\
    bind k select-pane -U\n\
    bind C-l select-pane -R\n\
    bind l select-pane -R\n\
    \n\
    # Vim-like splitting\n\
    bind s split-window -v\n\
    bind v split-window -h\n\
    \n\
    # Restore Ctrl-l and Ctrl-k\n\
    bind C-l send-keys '"'"'C-l'"'"'\n\
    bind C-k send-keys '"'"'C-k'"'"'\n\
    \n\
    run '"'"'~/.tmux/plugins/tpm/tpm'"'"'' > /root/.tmux.conf


# Install rust
RUN curl --proto '=https' --tlsv1.2 -sSf https://sh.rustup.rs | sh -s -- -y && \
    export PATH=$PATH:$HOME/.cargo/bin && \
    touch /lorax/proto/generate.proto

ENTRYPOINT ["/bin/bash"]