File tree Expand file tree Collapse file tree 1 file changed +2
-6
lines changed Expand file tree Collapse file tree 1 file changed +2
-6
lines changed Original file line number Diff line number Diff line change @@ -33,12 +33,8 @@ case ${GPU_ARCH_TYPE} in
33
33
DOCKER_TAG=cuda${GPU_ARCH_VERSION}
34
34
LEGACY_DOCKER_IMAGE=${DOCKER_REGISTRY} /pytorch/manylinux-cuda${GPU_ARCH_VERSION// ./ }
35
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} "
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"
42
38
;;
43
39
rocm)
44
40
TARGET=rocm_final
You can’t perform that action at this time.
0 commit comments