forked from IBM-CAMHub-Open/IBMPower
-
Notifications
You must be signed in to change notification settings - Fork 2
/
DockerfileNvidiaPAI
35 lines (28 loc) · 1010 Bytes
/
DockerfileNvidiaPAI
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
FROM nvidia/cuda-ppc64le:8.0-cudnn6-devel-ubuntu16.04
ENV POWERAI_REPO mldl-repo-local_4.0.0_ppc64el.deb
RUN apt-get update && apt-get install -y \
git \
wget \
ssh \
vim \
curl && \
apt-get clean
# import PowerAI repo
RUN cd /tmp && \
wget https://public.dhe.ibm.com/software/server/POWER/Linux/mldl/ubuntu/${POWERAI_REPO} && \
dpkg -i ${POWERAI_REPO} && \
rm ${POWERAI_REPO}
# install PowerAI
RUN apt-get update && \
apt-get install -y power-mldl && \
apt-get update && apt-get install -y openssh-server
RUN apt-get clean
RUN mkdir /var/run/sshd
RUN echo 'root:powerlinux' | chpasswd
RUN sed -i 's/PermitRootLogin prohibit-password/PermitRootLogin yes/' /etc/ssh/sshd_config
# SSH login fix. Otherwise user is kicked off after login
RUN sed 's@session\s*required\s*pam_loginuid.so@session optional pam_loginuid.so@g' -i /etc/pam.d/sshd
ENV NOTVISIBLE "in users profile"
RUN echo "export VISIBLE=now" >> /etc/profile
EXPOSE 22
CMD ["/usr/sbin/sshd", "-D"]