File tree Expand file tree Collapse file tree 2 files changed +12
-2
lines changed Expand file tree Collapse file tree 2 files changed +12
-2
lines changed Original file line number Diff line number Diff line change @@ -2,10 +2,17 @@ FROM nvcr.io/nvidia/tritonserver:20.12-py3
2
2
# NVIDIA Triton
3
3
4
4
ENV PYTHONUNBUFFERED=1 \
5
+ _CUDA_COMPAT_PATH="/usr/local/cuda/compat" \
5
6
PATH="/opt/tritonserver/bin:/usr/local/mpi/bin:/usr/local/nvidia/bin:/usr/local/cuda/bin:/usr/local/sbin:/usr/local/bin:/usr/sbin:/usr/bin:/sbin:/bin:/usr/local/ucx/bin" \
6
7
LD_LIBRARY_PATH="/usr/local/cuda/compat/lib:/usr/local/nvidia/lib:/usr/local/nvidia/lib64" \
7
8
LANG=C.UTF-8
8
9
10
+ RUN apt update && \
11
+ apt install -y --no-install-recommends \
12
+ ncurses-term \
13
+ unzip zlib1g-dev htop && \
14
+ ln -sf /usr/share/terminfo/x/xterm-color /usr/share/terminfo/x/xterm-256color
15
+
9
16
RUN update-alternatives --install /usr/bin/python python /usr/bin/python3 2
10
17
COPY ./service-defs /etc/backend.ai/service-defs
11
18
LABEL ai.backend.kernelspec="1" \
@@ -20,7 +27,7 @@ LABEL ai.backend.kernelspec="1" \
20
27
ai.backend.base-distro="ubuntu16.04" \
21
28
ai.backend.runtime-type="python" \
22
29
ai.backend.runtime-path="/usr/bin/python" \
23
- ai.backend.service-ports="grpc:http:8001,http:http: 8000,metric:http: 8002"
30
+ ai.backend.service-ports="triton-server:preopen:[ 8000,8001, 8002] "
24
31
25
32
WORKDIR /home/work
26
33
# vim: ft=dockerfile
Original file line number Diff line number Diff line change @@ -2,11 +2,14 @@ FROM nvcr.io/nvidia/tritonserver:20.12-py3-sdk
2
2
# NVIDIA Triton
3
3
4
4
ENV PYTHONUNBUFFERED=1 \
5
- PATH="/workspace/install/bin:/usr/local/nvidia/bin:/usr/local/cuda/bin:/usr/local/sbin:/usr/local/bin:/usr/sbin:/usr/bin:/sbin:/bin" \
5
+ PATH="/workspace/install/bin:/opt/tritonserver/bin:/ usr/local/nvidia/bin:/usr/local/cuda/bin:/usr/local/sbin:/usr/local/bin:/usr/sbin:/usr/bin:/sbin:/bin" \
6
6
LIBRARY_PATH="/usr/local/cuda/lib64/stubs" \
7
7
LD_LIBRARY_PATH="/workspace/install/lib:/usr/local/nvidia/lib:/usr/local/nvidia/lib64" \
8
8
LANG=C.UTF-8
9
9
10
+ RUN apt update ; apt install ncurses-term -y && \
11
+ ln -sf /usr/share/terminfo/x/xterm-color /usr/share/terminfo/x/xterm-256color
12
+
10
13
RUN update-alternatives --install /usr/bin/python python /usr/bin/python3 2
11
14
COPY ./service-defs /etc/backend.ai/service-defs
12
15
LABEL ai.backend.kernelspec="1" \
You can’t perform that action at this time.
0 commit comments