File tree Expand file tree Collapse file tree 1 file changed +5
-5
lines changed Expand file tree Collapse file tree 1 file changed +5
-5
lines changed Original file line number Diff line number Diff line change 7
7
# base python 3.7
8
8
FROM lablup/common-base:python3.7 as python-binary
9
9
10
- FROM nvidia/cuda:10.1 -cudnn7-devel-ubuntu16.04
10
+ FROM nvidia/cuda:10.0 -cudnn7-devel-ubuntu16.04
11
11
MAINTAINER Mario Cho "
[email protected] "
12
12
13
- ENV CUDA_VERSION 10.1.168
13
+ ENV CUDA_VERSION 10.0.130
14
14
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
17
17
18
18
RUN apt-get update -y && \
19
19
apt-mark unhold libnccl2 && \
@@ -39,7 +39,7 @@ ENV PYTHONUNBUFFERED=1 \
39
39
LANG=C.UTF-8
40
40
41
41
# 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 && \
43
43
ln -s /usr/lib/x86_64-linux-gnu/libcudnn.so.7.6.0 /usr/local/cuda/lib64/libcudnn.so && \
44
44
ldconfig
45
45
You can’t perform that action at this time.
0 commit comments