Skip to content

Commit 70f6c48

Browse files
authored
Update build_docker.sh (#1446)
Use [`nvidia/cuda:11.4.3-devel-centos7`](https://hub.docker.com/layers/nvidia/cuda/11.4.3-devel-centos7/images/sha256-e2201a4954dfd65958a6f5272cd80b968902789ff73f26151306907680356db8?context=explore) because `nvidia/cuda:10.2-devel-centos7` was deleted in accordance with [Nvidia's Container Support Policy](https://gitlab.com/nvidia/container-images/cuda/blob/master/doc/support-policy.md): > After a period of Six Months time, the EOL tags WILL BE DELETED from Docker Hub and Nvidia GPU Cloud (NGC). This deletion ensures unsupported tags (and image layers) are not left lying around for customers to continue using after they have long been abandoned. Also delete redundant DEVTOOLSET=7 clause
1 parent d9b2c42 commit 70f6c48

File tree

1 file changed

+2
-6
lines changed

1 file changed

+2
-6
lines changed

manywheel/build_docker.sh

Lines changed: 2 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -33,12 +33,8 @@ case ${GPU_ARCH_TYPE} in
3333
DOCKER_TAG=cuda${GPU_ARCH_VERSION}
3434
LEGACY_DOCKER_IMAGE=${DOCKER_REGISTRY}/pytorch/manylinux-cuda${GPU_ARCH_VERSION//./}
3535
# Keep this up to date with the minimum version of CUDA we currently support
36-
GPU_IMAGE=nvidia/cuda:10.2-devel-centos7
37-
DEVTOOLSET_VERSION="9"
38-
if [[ ${GPU_ARCH_VERSION:0:2} == "10" ]]; then
39-
DEVTOOLSET_VERSION="7"
40-
fi
41-
DOCKER_GPU_BUILD_ARG="--build-arg BASE_CUDA_VERSION=${GPU_ARCH_VERSION} --build-arg DEVTOOLSET_VERSION=${DEVTOOLSET_VERSION}"
36+
GPU_IMAGE=nvidia/cuda:11.4.3-devel-centos7
37+
DOCKER_GPU_BUILD_ARG="--build-arg BASE_CUDA_VERSION=${GPU_ARCH_VERSION} --build-arg DEVTOOLSET_VERSION=9"
4238
;;
4339
rocm)
4440
TARGET=rocm_final

0 commit comments

Comments
 (0)