Pytorch 1.4 for python 3.6 & CUDA 10

Author: hephaex

python-pytorch/Dockerfile.1.4-py36-cuda10

@@ -1,79 +1,17 @@
-FROM nvidia/cuda:10.0-cudnn7-devel-ubuntu18.04
-MAINTAINER Mario Cho "[email protected]"
-
-ENV LANG=C.UTF-8
-ENV PYTHONUNBUFFERED 1
-ENV CUDA_VERSION 10.0.130
-ENV CUDA_PKG_VERSION 10-0=$CUDA_VERSION-1
-ENV CUDNN_VERSION 7.6.4.38
-ENV NCCL_VERSION 2.4.8
-ENV LD_LIBRARY_PATH="/usr/local/cuda/lib64:/usr/local/nvidia/lib64" \
-    PATH="/usr/local/nvidia/bin:/usr/local/cuda/bin:/usr/local/sbin:/usr/local/bin:/usr/sbin:/usr/bin:/sbin:/bin"
-ENV DEBIAN_FRONTEND=noninteractive
+# Lablup/Python-PyTorch 1.4 Python 3.6, CUDA10.0
 
-RUN apt-get update && \
-    apt-get install -y \
-        ca-certificates \
-        wget curl git-core \
-        vim-tiny zip unzip \
-        python3 python3-pip \
-        libssl-dev \
-        libmpdec2 \
-        proj-bin libproj-dev \
-        libgeos-dev libgeos++-dev \
-        mime-support \
-        gcc g++ \
-    && \
-    apt-get clean && \
-    rm -rf /var/lib/apt/lists/
+FROM lablup/common-base:19.09-py36-cuda10
+MAINTAINER Mario Cho "[email protected]"
 
 ENV PYTHONUNBUFFERED=1 \
     LD_LIBRARY_PATH="/usr/local/cuda/lib64:/usr/local/nvidia/lib64" \
     PATH="/usr/local/nvidia/bin:/usr/local/cuda/bin:/usr/local/sbin:/usr/local/bin:/usr/sbin:/usr/bin:/sbin:/bin" \
     LANG=C.UTF-8
 
-# Install CUDA-10.0 + cuDNN 7.4
-RUN ln -s /usr/local/cuda-10.0 /usr/local/cuda && \
-    ln -s /usr/lib/x86_64-linux-gnu/libcudnn.so.7.6.0 /usr/local/cuda/lib64/libcudnn.so && \
-    ldconfig
-
-RUN apt-get update && apt-get install -y --no-install-recommends \
-        cuda-libraries-$CUDA_PKG_VERSION \
-        cuda-cublas-10-0 \
-        libnccl2=2.4.8-1+cuda10.0 && \
-    apt-mark hold libnccl2 && \
-    rm -rf /var/lib/apt/lists/*
-
-RUN curl https://bootstrap.pypa.io/get-pip.py | python3 && \
-    python3 -m pip install --no-cache-dir -U setuptools && \
-    python3 -m pip install --no-cache-dir wheel && \
-    python3 -m pip install --no-cache-dir pyzmq simplejson msgpack-python uvloop && \
-    python3 -m pip install --no-cache-dir aiozmq dataclasses tabulate && \
-    python3 -m pip install --no-cache-dir namedlist six "python-dateutil>=2" && \
-    python3 -m pip install --no-cache-dir h5py && \
-    python3 -m pip install --no-cache-dir Cython && \
-    python3 -m pip install --no-cache-dir numpy scipy && \
-    python3 -m pip install --no-cache-dir versioneer==0.17 && \
-    python3 -m pip install --no-cache-dir pyproj Cartopy==0.16 && \
-    python3 -m pip install --no-cache-dir matplotlib bokeh && \
-    python3 -m pip install --no-cache-dir pandas && \
-    python3 -m pip install --no-cache-dir seaborn && \
-    python3 -m pip install --no-cache-dir pillow && \
-    python3 -m pip install --no-cache-dir networkx cvxpy && \
-    python3 -m pip install --no-cache-dir scikit-learn scikit-image && \
-    python3 -m pip install --no-cache-dir pygments && \
-    python3 -m pip install --no-cache-dir torch==1.4 && \
-    python3 -m pip install --no-cache-dir torchvision && \
-    python3 -m pip install --no-cache-dir tensorwatch && \
-    python3 -m pip install --no-cache-dir ipython && \
-    python3 -m pip install --no-cache-dir jupyter && \
-    python3 -m pip install --no-cache-dir jupyterlab && \
-    rm -rf /root/.cache && \
-    rm -f /tmp/*.whl
 RUN update-alternatives --install /usr/bin/python python /usr/bin/python3 2
 
 # Install ipython kernelspec
-RUN python -m ipykernel install --display-name "PyTorch 1.4 on Python 3.6 (CUDA 10.0)" && \
+RUN python3 -m ipykernel install --display-name "PyTorch 1.4 on Python 3.6 (CUDA 10.0)" && \
     cat /usr/local/share/jupyter/kernels/python3/kernel.json
 
 # Backend.AI specifics
@@ -86,9 +24,8 @@ LABEL ai.backend.kernelspec="1" \
       ai.backend.resource.min.cuda.device=1 \
       ai.backend.resource.min.cuda.shares=0.1 \
       ai.backend.runtime-type="python" \
-      ai.backend.runtime-path="/usr/bin/python" \
-      ai.backend.service-ports="ipython:pty:3000,jupyter:http:8080,jupyterlab:http:8090,tensorboard:http:6006"
-
-COPY policy.yml /etc/backend.ai/jail/policy.yml
+      ai.backend.runtime-path="/usr/bin/python3" \
+      ai.backend.service-ports="ipython:pty:3000,jupyter:http:8070,jupyterlab:http:8090"
 
-# vim: ft=dockerfile
+WORKDIR /home/work
+# vim: ft=dockerfile