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

[MXNET-99] Upgrade to cuda 9.1 cudnn 7 #10108

Merged
merged 4 commits into from
Mar 21, 2018
Merged

[MXNET-99] Upgrade to cuda 9.1 cudnn 7 #10108

Show file tree
Hide file tree
Changes from 1 commit
Commits
Show all changes
4 commits
Select commit Hold shift + click to select a range
d3cac2f
Upgrade to cuda 9.1 cudnn 7
marcoabreu Mar 14, 2018
1bb80e6
Pin pylint version
marcoabreu Mar 20, 2018
f81a8fa
Remove aws cli installation
marcoabreu Mar 20, 2018
b76809f
Address review comments
marcoabreu Mar 20, 2018
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
Next Next commit
Upgrade to cuda 9.1 cudnn 7
  • Loading branch information
@marcoabreu
marcoabreu committed Mar 14, 2018
commit d3cac2f7e25090a7b6cfbfb93a662d9ed4cdc137
4 changes: 2 additions & 2 deletions Jenkinsfile
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -211,11 +211,11 @@ try {
}
}
},
'GPU: CUDA8.0+cuDNN5': {
'GPU: CUDA9.1+cuDNN7': {
node('mxnetlinux-cpu') {
ws('workspace/build-gpu') {
init_git()
sh "ci/build.py --build --platform ubuntu_build_cuda /work/runtime_functions.sh build_ubuntu_gpu_cuda8_cudnn5"
sh "ci/build.py --build --platform ubuntu_build_cuda /work/runtime_functions.sh build_ubuntu_gpu_cuda91_cudnn7"
pack_lib('gpu')
stash includes: 'build/cpp-package/example/test_score', name: 'cpp_test_score'
stash includes: 'build/cpp-package/example/test_optimizer', name: 'cpp_test_optimizer'
Expand Down
2 changes: 1 addition & 1 deletion ci/docker/Dockerfile.build.centos7_gpu
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -18,7 +18,7 @@
#
# Dockerfile to build and run MXNet on CentOS 7 for GPU

FROM nvidia/cuda:8.0-cudnn5-devel-centos7
FROM nvidia/cuda:9.1-cudnn7-devel-centos7

ARG USER_ID=0

Expand Down
4 changes: 2 additions & 2 deletions ci/docker/Dockerfile.build.ubuntu_build_cuda
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -21,7 +21,7 @@
# package generation, requiring the actual CUDA library to be
# present

FROM nvidia/cuda:8.0-cudnn5-devel
FROM nvidia/cuda:9.1-cudnn7-devel

ARG USER_ID=0

Expand Down Expand Up @@ -54,4 +54,4 @@ RUN /work/ubuntu_nvidia.sh
COPY runtime_functions.sh /work/

WORKDIR /work/mxnet
ENV LD_LIBRARY_PATH=${LD_LIBRARY_PATH}:/usr/local/lib
ENV LD_LIBRARY_PATH=${LD_LIBRARY_PATH}:/usr/local/lib
2 changes: 1 addition & 1 deletion ci/docker/Dockerfile.build.ubuntu_gpu
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -18,7 +18,7 @@
#
# Dockerfile to run MXNet on Ubuntu 16.04 for CPU

FROM nvidia/cuda:8.0-cudnn5-devel
FROM nvidia/cuda:9.1-cudnn7-devel

ARG USER_ID=0

Expand Down
5 changes: 1 addition & 4 deletions ci/docker/install/ubuntu_nvidia.sh
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -17,13 +17,10 @@
# specific language governing permissions and limitations
# under the License.

# build and install are separated so changes to build don't invalidate
# the whole docker cache for the image

set -ex
apt install -y software-properties-common
add-apt-repository -y ppa:graphics-drivers
# Retrieve ppa:graphics-drivers and install nvidia-drivers.
# Note: DEBIAN_FRONTEND required to skip the interactive setup steps
apt update
DEBIAN_FRONTEND=noninteractive apt install -y --no-install-recommends cuda-8-0
DEBIAN_FRONTEND=noninteractive apt install -y --no-install-recommends cuda-9-1
2 changes: 1 addition & 1 deletion ci/docker/runtime_functions.sh
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -256,7 +256,7 @@ build_ubuntu_gpu_mkldnn() {
-j$(nproc)
}

build_ubuntu_gpu_cuda8_cudnn5() {
build_ubuntu_gpu_cuda91_cudnn7() {
set -ex
make \
DEV=1 \
Expand Down