1# -*- mode: dockerfile -*-
2# Licensed to the Apache Software Foundation (ASF) under one
3# or more contributor license agreements.  See the NOTICE file
4# distributed with this work for additional information
5# regarding copyright ownership.  The ASF licenses this file
6# to you under the Apache License, Version 2.0 (the
7# "License"); you may not use this file except in compliance
8# with the License.  You may obtain a copy of the License at
9#
10#   http://www.apache.org/licenses/LICENSE-2.0
11#
12# Unless required by applicable law or agreed to in writing,
13# software distributed under the License is distributed on an
14# "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
15# KIND, either express or implied.  See the License for the
16# specific language governing permissions and limitations
17# under the License.
18#
19# Dockerfile to run MXNet on Ubuntu 18.04 for GPU
20
21FROM nvidia/cuda:10.2-devel-ubuntu18.04
22
23WORKDIR /work/deps
24
25COPY install/ubuntu_core.sh /work/
26RUN /work/ubuntu_core.sh
27
28COPY install/deb_ubuntu_ccache.sh /work/
29RUN /work/deb_ubuntu_ccache.sh
30
31COPY install/ubuntu_python.sh /work/
32COPY install/requirements /work/
33RUN /work/ubuntu_python.sh
34
35COPY install/ubuntu_scala.sh /work/
36COPY install/sbt.gpg /work/
37RUN /work/ubuntu_scala.sh
38
39COPY install/ubuntu_r.sh /work/
40RUN /work/ubuntu_r.sh
41
42COPY install/ubuntu_perl.sh /work/
43RUN /work/ubuntu_perl.sh
44
45COPY install/ubuntu_clang.sh /work/
46RUN /work/ubuntu_clang.sh
47
48COPY install/ubuntu_tvm.sh /work/
49RUN /work/ubuntu_tvm.sh
50
51COPY install/ubuntu_llvm.sh /work/
52RUN /work/ubuntu_llvm.sh
53
54COPY install/ubuntu_caffe.sh /work/
55RUN /work/ubuntu_caffe.sh
56
57COPY install/ubuntu_onnx.sh /work/
58RUN /work/ubuntu_onnx.sh
59
60COPY install/ubuntu_docs.sh /work/
61COPY install/requirements /work/
62RUN /work/ubuntu_docs.sh
63
64COPY install/ubuntu_tutorials.sh /work/
65RUN /work/ubuntu_tutorials.sh
66
67ENV CUDA_VERSION=10.2.89
68ENV CUDNN_VERSION=8.0.4.30
69COPY install/ubuntu_cudnn.sh /work/
70RUN /work/ubuntu_cudnn.sh
71
72# Always last
73ARG USER_ID=0
74ARG GROUP_ID=0
75COPY install/ubuntu_adduser.sh /work/
76RUN /work/ubuntu_adduser.sh
77
78COPY runtime_functions.sh /work/
79
80WORKDIR /work/mxnet
81