-
Notifications
You must be signed in to change notification settings - Fork 49
/
Copy pathDockerfile
35 lines (27 loc) · 1.04 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
FROM nvcr.io/nvidia/tensorflow:19.09-py3
MAINTAINER [email protected]
RUN sed -i 's/archive.ubuntu.com/mirrors.ustc.edu.cn/g' /etc/apt/sources.list
RUN apt update -y &&\
apt install -y libcurl3 &&\
apt install -y lsb-core &&\
apt clean
# Install RDMA driver
WORKDIR /tmp
RUN wget http://content.mellanox.com/ofed/MLNX_OFED-4.7-1.0.0.1/MLNX_OFED_LINUX-4.7-1.0.0.1-ubuntu18.04-x86_64.tgz
# COPY MLNX_OFED_LINUX-4.7-1.0.0.1-ubuntu18.04-x86_64.tgz .
RUN tar xvf MLNX_OFED_LINUX-4.7-1.0.0.1-ubuntu18.04-x86_64.tgz &&\
cd MLNX_OFED_LINUX-4.7-1.0.0.1-ubuntu18.04-x86_64 &&\
./mlnxofedinstall --user-space-only --without-fw-update --all --force -q &&\
cd /tmp && rm -rf *
# Install Orion Client
WORKDIR /opt
COPY install-client-10.1 .
RUN chmod +x install-client-10.1 && ./install-client-10.1 -q
ENV LD_LIBRARY_PATH=/usr/lib/orion
# Set default ORION_VGPU for each process requesting vgpu resources
ENV ORION_VGPU 1
ENV ORION_GMEM 4096
WORKDIR /root
# A toy RNN model (sequence classification)
COPY demo_story_RNN_code.py .
CMD ["/bin/bash"]