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

[BUGFIX]correcting cuda 11.2 image name in CI and CD #19960

Merged
merged 1 commit into from
Feb 26, 2021
Merged

[BUGFIX]correcting cuda 11.2 image name in CI and CD #19960

Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
1 commit
Select commit
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Jump to
Jump to file
Failed to load files.
Diff view
Diff view
2 changes: 1 addition & 1 deletion cd/utils/mxnet_base_image.sh
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -34,7 +34,7 @@ case ${mxnet_variant} in
echo "nvidia/cuda:11.0-cudnn8-runtime-ubuntu16.04"
;;
cu112*)
echo "nvidia/cuda:11.2-cudnn8-runtime-ubuntu16.04"
echo "nvidia/cuda:11.2.1-cudnn8-runtime-ubuntu16.04"
;;
cpu)
echo "ubuntu:16.04"
Expand Down
2 changes: 1 addition & 1 deletion ci/docker/Dockerfile.build.ubuntu_gpu_cu112
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 GPU

FROM nvidia/cuda:11.2-cudnn8-devel-ubuntu16.04
FROM nvidia/cuda:11.2.1-cudnn8-devel-ubuntu16.04

WORKDIR /work/deps

Expand Down