File tree Expand file tree Collapse file tree 1 file changed +30
-0
lines changed Expand file tree Collapse file tree 1 file changed +30
-0
lines changed Original file line number Diff line number Diff line change
1
+ FROM nvcr.io/nvidia/tritonserver:22.07-py3-sdk
2
+ # NVIDIA Triton
3
+
4
+ ENV PYTHONUNBUFFERED=1 \
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
+ LIBRARY_PATH="/usr/local/cuda/lib64/stubs" \
7
+ LD_LIBRARY_PATH="/workspace/install/lib:/usr/local/nvidia/lib:/usr/local/nvidia/lib64" \
8
+ LANG=C.UTF-8
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
+
13
+ RUN update-alternatives --install /usr/bin/python python /usr/bin/python3 2
14
+ COPY ./service-defs /etc/backend.ai/service-defs
15
+ LABEL ai.backend.kernelspec="1" \
16
+ ai.backend.envs.corecount="OPENBLAS_NUM_THREADS,OMP_NUM_THREADS,NPROC" \
17
+ ai.backend.features="batch query uid-match user-input" \
18
+ ai.backend.base-distro="ubuntu16.04" \
19
+ ai.backend.accelerators="cuda" \
20
+ ai.backend.resource.min.cpu="1" \
21
+ ai.backend.resource.min.mem="1g" \
22
+ ai.backend.resource.min.cuda.device=1 \
23
+ ai.backend.resource.min.cuda.shares=0.1 \
24
+ ai.backend.base-distro="ubuntu16.04" \
25
+ ai.backend.runtime-type="python" \
26
+ ai.backend.runtime-path="/usr/bin/python" \
27
+ ai.backend.service-ports=""
28
+
29
+ WORKDIR /home/work
30
+ # vim: ft=dockerfile
You can’t perform that action at this time.
0 commit comments