We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent d9b2c42 commit 70f6c48Copy full SHA for 70f6c48
manywheel/build_docker.sh
@@ -33,12 +33,8 @@ case ${GPU_ARCH_TYPE} in
33
DOCKER_TAG=cuda${GPU_ARCH_VERSION}
34
LEGACY_DOCKER_IMAGE=${DOCKER_REGISTRY}/pytorch/manylinux-cuda${GPU_ARCH_VERSION//./}
35
# 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}"
+ GPU_IMAGE=nvidia/cuda:11.4.3-devel-centos7
+ DOCKER_GPU_BUILD_ARG="--build-arg BASE_CUDA_VERSION=${GPU_ARCH_VERSION} --build-arg DEVTOOLSET_VERSION=9"
42
;;
43
rocm)
44
TARGET=rocm_final
0 commit comments