This repository was archived by the owner on Nov 17, 2023. It is now read-only.
File tree Expand file tree Collapse file tree 2 files changed +37
-1
lines changed
Expand file tree Collapse file tree 2 files changed +37
-1
lines changed Original file line number Diff line number Diff line change 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
2325WORKDIR /work/deps
2426
@@ -28,6 +30,8 @@ COPY install/centos7_ccache.sh /work/
2830RUN /work/centos7_ccache.sh
2931COPY install/centos7_python.sh /work/
3032RUN /work/centos7_python.sh
33+ COPY install/centos7_cudnn.sh /work/
34+ RUN /work/centos7_cudnn.sh
3135
3236ARG USER_ID=0
3337COPY install/centos7_adduser.sh /work/
Original file line number Diff line number Diff line change 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+
You can’t perform that action at this time.
0 commit comments