Skip to content

Commit ec84721

Browse files
committed
Base kernel for Python 3.7
1 parent 191c0c7 commit ec84721

File tree

1 file changed

+5
-5
lines changed

1 file changed

+5
-5
lines changed

commons/Dockerfile.base.19.08-py37-cuda10

Lines changed: 5 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -7,13 +7,13 @@
77
# base python 3.7
88
FROM lablup/common-base:python3.7 as python-binary
99

10-
FROM nvidia/cuda:10.1-cudnn7-devel-ubuntu16.04
10+
FROM nvidia/cuda:10.0-cudnn7-devel-ubuntu16.04
1111
MAINTAINER Mario Cho "[email protected]"
1212

13-
ENV CUDA_VERSION 10.1.168
13+
ENV CUDA_VERSION 10.0.130
1414
ENV CUDA_PKG_VERSION 10-1=$CUDA_VERSION-1
15-
ENV CUDNN_VERSION 7.6.2.24
16-
ENV NCCL_VERSION 2.4.7
15+
ENV CUDNN_VERSION 7.6.4.38
16+
ENV NCCL_VERSION 2.4.8
1717

1818
RUN apt-get update -y && \
1919
apt-mark unhold libnccl2 && \
@@ -39,7 +39,7 @@ ENV PYTHONUNBUFFERED=1 \
3939
LANG=C.UTF-8
4040

4141
# Install CUDA-10.1 + cuDNN 7.6.0
42-
RUN ln -s /usr/local/cuda-10.1 /usr/local/cuda && \
42+
RUN ln -s /usr/local/cuda-10.0 /usr/local/cuda && \
4343
ln -s /usr/lib/x86_64-linux-gnu/libcudnn.so.7.6.0 /usr/local/cuda/lib64/libcudnn.so && \
4444
ldconfig
4545

0 commit comments

Comments
 (0)