Skip to content

Commit da72436

Browse files
committed
MXNet 1.5 for python 3.6 & CUDA10
1 parent 0153f52 commit da72436

File tree

1 file changed

+31
-0
lines changed

1 file changed

+31
-0
lines changed
Lines changed: 31 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,31 @@
1+
# Lablup/Python-mxnet 1.5 Python 3.6, CUDA10.0
2+
3+
FROM lablup/common-base:19.09-py36-cuda10
4+
MAINTAINER Mario Cho "[email protected]"
5+
6+
ENV PYTHONUNBUFFERED=1 \
7+
LD_LIBRARY_PATH="/usr/local/cuda/lib64:/usr/local/nvidia/lib64" \
8+
PATH="/usr/local/nvidia/bin:/usr/local/cuda/bin:/usr/local/sbin:/usr/local/bin:/usr/sbin:/usr/bin:/sbin:/bin" \
9+
LANG=C.UTF-8
10+
11+
RUN update-alternatives --install /usr/bin/python python /usr/bin/python3 2
12+
13+
# Install ipython kernelspec
14+
RUN python3 -m ipykernel install --display-name "mxnet 1.5 on Python 3.6 (CUDA 10.0)" && \
15+
cat /usr/local/share/jupyter/kernels/python3/kernel.json
16+
17+
# Backend.AI specifics
18+
LABEL ai.backend.kernelspec="1" \
19+
ai.backend.envs.corecount="OPENBLAS_NUM_THREADS,OMP_NUM_THREADS,NPROC" \
20+
ai.backend.features="batch query uid-match user-input" \
21+
ai.backend.base-distro="ubuntu16.04" \
22+
ai.backend.resource.min.cpu="1" \
23+
ai.backend.resource.min.mem="1g" \
24+
ai.backend.resource.min.cuda.device=1 \
25+
ai.backend.resource.min.cuda.shares=0.1 \
26+
ai.backend.runtime-type="python" \
27+
ai.backend.runtime-path="/usr/bin/python3" \
28+
ai.backend.service-ports="ipython:pty:3000,jupyter:http:8070,jupyterlab:http:8090"
29+
30+
WORKDIR /home/work
31+
# vim: ft=dockerfile

0 commit comments

Comments
 (0)