Skip to content
This repository was archived by the owner on Nov 17, 2023. It is now read-only.

Commit 7c0b80b

Browse files
committed
Updates CentOS 7 GPU CI image base image to cuda10-devel and manually installs cuDNN version 7.3.1.20
1 parent 4cdd9c0 commit 7c0b80b

File tree

2 files changed

+37
-1
lines changed

2 files changed

+37
-1
lines changed

ci/docker/Dockerfile.build.centos7_gpu

Lines changed: 5 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -18,7 +18,9 @@
1818
#
1919
# Dockerfile to build and run MXNet on CentOS 7 for GPU
2020

21-
FROM nvidia/cuda:9.1-cudnn7-devel-centos7
21+
FROM nvidia/cuda:10.0-devel-centos7
22+
23+
ENV CUDNN_VERSION=7.3.1.20
2224

2325
WORKDIR /work/deps
2426

@@ -28,6 +30,8 @@ COPY install/centos7_ccache.sh /work/
2830
RUN /work/centos7_ccache.sh
2931
COPY install/centos7_python.sh /work/
3032
RUN /work/centos7_python.sh
33+
COPY install/centos7_cudnn.sh /work/
34+
RUN /work/centos7_cudnn.sh
3135

3236
ARG USER_ID=0
3337
COPY install/centos7_adduser.sh /work/

ci/docker/install/centos7_cudnn.sh

Lines changed: 32 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,32 @@
1+
#!/usr/bin/env bash
2+
3+
# Licensed to the Apache Software Foundation (ASF) under one
4+
# or more contributor license agreements. See the NOTICE file
5+
# distributed with this work for additional information
6+
# regarding copyright ownership. The ASF licenses this file
7+
# to you under the Apache License, Version 2.0 (the
8+
# "License"); you may not use this file except in compliance
9+
# with the License. You may obtain a copy of the License at
10+
#
11+
# http://www.apache.org/licenses/LICENSE-2.0
12+
#
13+
# Unless required by applicable law or agreed to in writing,
14+
# software distributed under the License is distributed on an
15+
# "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
16+
# KIND, either express or implied. See the License for the
17+
# specific language governing permissions and limitations
18+
# under the License.
19+
20+
# build and install are separated so changes to build don't invalidate
21+
# the whole docker cache for the image
22+
23+
set -ex
24+
25+
# Multipackage installation does not fail in yum
26+
CUDNN_DOWNLOAD_SUM=4e15a323f2edffa928b4574f696fc0e449a32e6bc35c9ccb03a47af26c2de3fa
27+
curl -fsSL http://developer.download.nvidia.com/compute/redist/cudnn/v7.3.1/cudnn-10.0-linux-x64-v7.3.1.20.tgz -O
28+
echo "$CUDNN_DOWNLOAD_SUM cudnn-10.0-linux-x64-v7.3.1.20.tgz" | sha256sum -c -
29+
tar --no-same-owner -xzf cudnn-10.0-linux-x64-v7.3.1.20.tgz -C /usr/local
30+
rm cudnn-10.0-linux-x64-v7.3.1.20.tgz
31+
ldconfig
32+

0 commit comments

Comments
 (0)