1616# See the License for the specific language governing permissions and
1717# limitations under the License.
1818#
19- # Base unbuntu 16.04 image
20- FROM nvidia/cuda:9.0-devel-ubuntu16.04
19+ FROM nvidia/cuda:10.2-base-ubuntu18.04
2120
2221MAINTAINER singa dev@singa.apache.org
2322
24- # install dependencies
25- RUN apt-get update \
26- && apt-get install -y --no-install-recommends subversion git wget openssh-server bzip2 \
27- && apt-get clean && apt-get autoremove && apt-get autoclean \
28- && rm -rf /var/lib/apt/lists/*
29-
30- # install conda
31- RUN wget --no-check-certificate https://repo.continuum.io/miniconda/Miniconda3-py37_4.8.2-Linux-x86_64.sh -O miniconda.sh;
32- RUN bash miniconda.sh -b -p /root/miniconda
33- ENV PATH /root/miniconda/bin:${PATH}
34- RUN conda config --set always_yes yes --set changeps1 no
35- RUN conda config --add channels conda-forge
36- RUN conda config --add channels nusdbsystem
37- RUN conda install -c conda-forge -c nusdbsystem singa-gpu
38- RUN conda install -c conda-forge sphinx
39- RUN conda install -c conda-forge sphinx_rtd_theme
40- RUN conda install -c conda-forge recommonmark
41-
42-
43- # config ssh service
44- RUN mkdir /var/run/sshd \
45- && echo 'root:singa' | chpasswd \
46- && sed -ri 's/^#?PermitRootLogin\s +.*/PermitRootLogin yes/' /etc/ssh/sshd_config \
47- && sed -ri 's/UsePAM yes/#UsePAM yes/g' /etc/ssh/sshd_config \
48- && mkdir /root/.ssh
49-
50- # add conda bin path to login or non-login shell
51- RUN echo PATH=$PATH:/root/miniconda/bin >> /etc/profile
52-
53-
54- EXPOSE 22
55-
56- CMD ["/usr/sbin/sshd" , "-D" ]
23+ RUN apt-get update && apt-get install -y --no-install-recommends python3 python3-pip
24+ RUN pip3 install --upgrade pip
25+ RUN pip install singa -f http://singa.apache.org/docs/next/wheel-gpu.html --trusted-host singa.apache.org
0 commit comments