diff --git a/.github/scripts/generate_binary_build_matrix.py b/.github/scripts/generate_binary_build_matrix.py index 70c69fde929fa..a32cba2cfa7bd 100644 --- a/.github/scripts/generate_binary_build_matrix.py +++ b/.github/scripts/generate_binary_build_matrix.py @@ -31,7 +31,10 @@ def arch_type(arch_version: str) -> str: WHEEL_CONTAINER_IMAGES = { **{ gpu_arch: f"pytorch/manylinux-builder:cuda{gpu_arch}" - for gpu_arch in CUDA_ARCHES + for gpu_arch in CUDA_ARCHES if gpu_arch == "11.6" + }, + **{ + "11.7": "tousif111/manylinux-builder:cuda11.7" }, **{ gpu_arch: f"pytorch/manylinux-builder:rocm{gpu_arch}" @@ -219,7 +222,8 @@ def generate_wheels_matrix(os: str, "container_image": WHEEL_CONTAINER_IMAGES[arch_version], "package_type": package_type, "pytorch_extra_install_requirements": - "nvidia-cuda-runtime-cu11; platform_system == 'Linux' | " + "nvidia-cuda-nvrtc-cu11==11.7.99; platform_system == 'Linux' | " + "nvidia-cuda-runtime-cu11==11.7.99; platform_system == 'Linux' | " "nvidia-cuda-cupti-cu11==11.7.101; platform_system == 'Linux' | " "nvidia-cudnn-cu11==8.5.0.96; platform_system == 'Linux' | " "nvidia-cublas-cu11==11.10.3.66; platform_system == 'Linux' | " @@ -227,6 +231,7 @@ def generate_wheels_matrix(os: str, "nvidia-curand-cu11==10.2.10.91; platform_system == 'Linux' | " "nvidia-cusolver-cu11==11.4.0.1; platform_system == 'Linux' | " "nvidia-cusparse-cu11==11.7.4.91; platform_system == 'Linux' | " + "nvidia-nccl-cu11==2.14.3; platform_system == 'Linux' | " "nvidia-nvtx-cu11==11.7.91; platform_system == 'Linux'", "build_name": f"{package_type}-py{python_version}-{gpu_arch_type}{gpu_arch_version}-with-pypi-cudnn" diff --git a/.github/workflows/_binary-build-linux.yml b/.github/workflows/_binary-build-linux.yml index a8c533070c8bb..39099f82135d1 100644 --- a/.github/workflows/_binary-build-linux.yml +++ b/.github/workflows/_binary-build-linux.yml @@ -160,7 +160,7 @@ jobs: with: ref: main submodules: recursive - repository: pytorch/builder + repository: syed-ahmed/builder path: builder quiet-checkout: true - name: Clean pytorch/builder checkout diff --git a/.github/workflows/generated-linux-binary-manywheel-nightly.yml b/.github/workflows/generated-linux-binary-manywheel-nightly.yml index a624a817581d6..9dcafcbb7f313 100644 --- a/.github/workflows/generated-linux-binary-manywheel-nightly.yml +++ b/.github/workflows/generated-linux-binary-manywheel-nightly.yml @@ -165,11 +165,11 @@ jobs: DESIRED_CUDA: cu117 GPU_ARCH_VERSION: 11.7 GPU_ARCH_TYPE: cuda - DOCKER_IMAGE: pytorch/manylinux-builder:cuda11.7 + DOCKER_IMAGE: tousif111/manylinux-builder:cuda11.7 DESIRED_PYTHON: "3.7" build_name: manywheel-py3_7-cuda11_7-with-pypi-cudnn build_environment: linux-binary-manywheel - PYTORCH_EXTRA_INSTALL_REQUIREMENTS: nvidia-cuda-runtime-cu11; platform_system == 'Linux' | nvidia-cuda-cupti-cu11==11.7.101; platform_system == 'Linux' | nvidia-cudnn-cu11==8.5.0.96; platform_system == 'Linux' | nvidia-cublas-cu11==11.10.3.66; platform_system == 'Linux' | nvidia-cufft-cu11==10.9.0.58; platform_system == 'Linux' | nvidia-curand-cu11==10.2.10.91; platform_system == 'Linux' | nvidia-cusolver-cu11==11.4.0.1; platform_system == 'Linux' | nvidia-cusparse-cu11==11.7.4.91; platform_system == 'Linux' | nvidia-nvtx-cu11==11.7.91; platform_system == 'Linux' + PYTORCH_EXTRA_INSTALL_REQUIREMENTS: nvidia-cuda-nvrtc-cu11==11.7.99; platform_system == 'Linux' | nvidia-cuda-runtime-cu11==11.7.99; platform_system == 'Linux' | nvidia-cuda-cupti-cu11==11.7.101; platform_system == 'Linux' | nvidia-cudnn-cu11==8.5.0.96; platform_system == 'Linux' | nvidia-cublas-cu11==11.10.3.66; platform_system == 'Linux' | nvidia-cufft-cu11==10.9.0.58; platform_system == 'Linux' | nvidia-curand-cu11==10.2.10.91; platform_system == 'Linux' | nvidia-cusolver-cu11==11.4.0.1; platform_system == 'Linux' | nvidia-cusparse-cu11==11.7.4.91; platform_system == 'Linux' | nvidia-nccl-cu11==2.14.3; platform_system == 'Linux' | nvidia-nvtx-cu11==11.7.91; platform_system == 'Linux' secrets: github-token: ${{ secrets.GITHUB_TOKEN }} @@ -186,7 +186,7 @@ jobs: DESIRED_CUDA: cu117 GPU_ARCH_VERSION: 11.7 GPU_ARCH_TYPE: cuda - DOCKER_IMAGE: pytorch/manylinux-builder:cuda11.7 + DOCKER_IMAGE: tousif111/manylinux-builder:cuda11.7 DESIRED_PYTHON: "3.7" build_name: manywheel-py3_7-cuda11_7-with-pypi-cudnn build_environment: linux-binary-manywheel @@ -205,7 +205,7 @@ jobs: DESIRED_CUDA: cu117 GPU_ARCH_VERSION: 11.7 GPU_ARCH_TYPE: cuda - DOCKER_IMAGE: pytorch/manylinux-builder:cuda11.7 + DOCKER_IMAGE: tousif111/manylinux-builder:cuda11.7 DESIRED_PYTHON: "3.7" build_name: manywheel-py3_7-cuda11_7-with-pypi-cudnn secrets: @@ -226,7 +226,7 @@ jobs: DESIRED_CUDA: cu117 GPU_ARCH_VERSION: 11.7 GPU_ARCH_TYPE: cuda - DOCKER_IMAGE: pytorch/manylinux-builder:cuda11.7 + DOCKER_IMAGE: tousif111/manylinux-builder:cuda11.7 DESIRED_PYTHON: "3.7" build_name: manywheel-py3_7-cuda11_7 build_environment: linux-binary-manywheel @@ -246,7 +246,7 @@ jobs: DESIRED_CUDA: cu117 GPU_ARCH_VERSION: 11.7 GPU_ARCH_TYPE: cuda - DOCKER_IMAGE: pytorch/manylinux-builder:cuda11.7 + DOCKER_IMAGE: tousif111/manylinux-builder:cuda11.7 DESIRED_PYTHON: "3.7" build_name: manywheel-py3_7-cuda11_7 build_environment: linux-binary-manywheel @@ -265,7 +265,7 @@ jobs: DESIRED_CUDA: cu117 GPU_ARCH_VERSION: 11.7 GPU_ARCH_TYPE: cuda - DOCKER_IMAGE: pytorch/manylinux-builder:cuda11.7 + DOCKER_IMAGE: tousif111/manylinux-builder:cuda11.7 DESIRED_PYTHON: "3.7" build_name: manywheel-py3_7-cuda11_7 secrets: @@ -697,11 +697,11 @@ jobs: DESIRED_CUDA: cu117 GPU_ARCH_VERSION: 11.7 GPU_ARCH_TYPE: cuda - DOCKER_IMAGE: pytorch/manylinux-builder:cuda11.7 + DOCKER_IMAGE: tousif111/manylinux-builder:cuda11.7 DESIRED_PYTHON: "3.8" build_name: manywheel-py3_8-cuda11_7-with-pypi-cudnn build_environment: linux-binary-manywheel - PYTORCH_EXTRA_INSTALL_REQUIREMENTS: nvidia-cuda-runtime-cu11; platform_system == 'Linux' | nvidia-cuda-cupti-cu11==11.7.101; platform_system == 'Linux' | nvidia-cudnn-cu11==8.5.0.96; platform_system == 'Linux' | nvidia-cublas-cu11==11.10.3.66; platform_system == 'Linux' | nvidia-cufft-cu11==10.9.0.58; platform_system == 'Linux' | nvidia-curand-cu11==10.2.10.91; platform_system == 'Linux' | nvidia-cusolver-cu11==11.4.0.1; platform_system == 'Linux' | nvidia-cusparse-cu11==11.7.4.91; platform_system == 'Linux' | nvidia-nvtx-cu11==11.7.91; platform_system == 'Linux' + PYTORCH_EXTRA_INSTALL_REQUIREMENTS: nvidia-cuda-nvrtc-cu11==11.7.99; platform_system == 'Linux' | nvidia-cuda-runtime-cu11==11.7.99; platform_system == 'Linux' | nvidia-cuda-cupti-cu11==11.7.101; platform_system == 'Linux' | nvidia-cudnn-cu11==8.5.0.96; platform_system == 'Linux' | nvidia-cublas-cu11==11.10.3.66; platform_system == 'Linux' | nvidia-cufft-cu11==10.9.0.58; platform_system == 'Linux' | nvidia-curand-cu11==10.2.10.91; platform_system == 'Linux' | nvidia-cusolver-cu11==11.4.0.1; platform_system == 'Linux' | nvidia-cusparse-cu11==11.7.4.91; platform_system == 'Linux' | nvidia-nccl-cu11==2.14.3; platform_system == 'Linux' | nvidia-nvtx-cu11==11.7.91; platform_system == 'Linux' secrets: github-token: ${{ secrets.GITHUB_TOKEN }} @@ -718,7 +718,7 @@ jobs: DESIRED_CUDA: cu117 GPU_ARCH_VERSION: 11.7 GPU_ARCH_TYPE: cuda - DOCKER_IMAGE: pytorch/manylinux-builder:cuda11.7 + DOCKER_IMAGE: tousif111/manylinux-builder:cuda11.7 DESIRED_PYTHON: "3.8" build_name: manywheel-py3_8-cuda11_7-with-pypi-cudnn build_environment: linux-binary-manywheel @@ -737,7 +737,7 @@ jobs: DESIRED_CUDA: cu117 GPU_ARCH_VERSION: 11.7 GPU_ARCH_TYPE: cuda - DOCKER_IMAGE: pytorch/manylinux-builder:cuda11.7 + DOCKER_IMAGE: tousif111/manylinux-builder:cuda11.7 DESIRED_PYTHON: "3.8" build_name: manywheel-py3_8-cuda11_7-with-pypi-cudnn secrets: @@ -758,7 +758,7 @@ jobs: DESIRED_CUDA: cu117 GPU_ARCH_VERSION: 11.7 GPU_ARCH_TYPE: cuda - DOCKER_IMAGE: pytorch/manylinux-builder:cuda11.7 + DOCKER_IMAGE: tousif111/manylinux-builder:cuda11.7 DESIRED_PYTHON: "3.8" build_name: manywheel-py3_8-cuda11_7 build_environment: linux-binary-manywheel @@ -778,7 +778,7 @@ jobs: DESIRED_CUDA: cu117 GPU_ARCH_VERSION: 11.7 GPU_ARCH_TYPE: cuda - DOCKER_IMAGE: pytorch/manylinux-builder:cuda11.7 + DOCKER_IMAGE: tousif111/manylinux-builder:cuda11.7 DESIRED_PYTHON: "3.8" build_name: manywheel-py3_8-cuda11_7 build_environment: linux-binary-manywheel @@ -797,7 +797,7 @@ jobs: DESIRED_CUDA: cu117 GPU_ARCH_VERSION: 11.7 GPU_ARCH_TYPE: cuda - DOCKER_IMAGE: pytorch/manylinux-builder:cuda11.7 + DOCKER_IMAGE: tousif111/manylinux-builder:cuda11.7 DESIRED_PYTHON: "3.8" build_name: manywheel-py3_8-cuda11_7 secrets: @@ -1229,11 +1229,11 @@ jobs: DESIRED_CUDA: cu117 GPU_ARCH_VERSION: 11.7 GPU_ARCH_TYPE: cuda - DOCKER_IMAGE: pytorch/manylinux-builder:cuda11.7 + DOCKER_IMAGE: tousif111/manylinux-builder:cuda11.7 DESIRED_PYTHON: "3.9" build_name: manywheel-py3_9-cuda11_7-with-pypi-cudnn build_environment: linux-binary-manywheel - PYTORCH_EXTRA_INSTALL_REQUIREMENTS: nvidia-cuda-runtime-cu11; platform_system == 'Linux' | nvidia-cuda-cupti-cu11==11.7.101; platform_system == 'Linux' | nvidia-cudnn-cu11==8.5.0.96; platform_system == 'Linux' | nvidia-cublas-cu11==11.10.3.66; platform_system == 'Linux' | nvidia-cufft-cu11==10.9.0.58; platform_system == 'Linux' | nvidia-curand-cu11==10.2.10.91; platform_system == 'Linux' | nvidia-cusolver-cu11==11.4.0.1; platform_system == 'Linux' | nvidia-cusparse-cu11==11.7.4.91; platform_system == 'Linux' | nvidia-nvtx-cu11==11.7.91; platform_system == 'Linux' + PYTORCH_EXTRA_INSTALL_REQUIREMENTS: nvidia-cuda-nvrtc-cu11==11.7.99; platform_system == 'Linux' | nvidia-cuda-runtime-cu11==11.7.99; platform_system == 'Linux' | nvidia-cuda-cupti-cu11==11.7.101; platform_system == 'Linux' | nvidia-cudnn-cu11==8.5.0.96; platform_system == 'Linux' | nvidia-cublas-cu11==11.10.3.66; platform_system == 'Linux' | nvidia-cufft-cu11==10.9.0.58; platform_system == 'Linux' | nvidia-curand-cu11==10.2.10.91; platform_system == 'Linux' | nvidia-cusolver-cu11==11.4.0.1; platform_system == 'Linux' | nvidia-cusparse-cu11==11.7.4.91; platform_system == 'Linux' | nvidia-nccl-cu11==2.14.3; platform_system == 'Linux' | nvidia-nvtx-cu11==11.7.91; platform_system == 'Linux' secrets: github-token: ${{ secrets.GITHUB_TOKEN }} @@ -1250,7 +1250,7 @@ jobs: DESIRED_CUDA: cu117 GPU_ARCH_VERSION: 11.7 GPU_ARCH_TYPE: cuda - DOCKER_IMAGE: pytorch/manylinux-builder:cuda11.7 + DOCKER_IMAGE: tousif111/manylinux-builder:cuda11.7 DESIRED_PYTHON: "3.9" build_name: manywheel-py3_9-cuda11_7-with-pypi-cudnn build_environment: linux-binary-manywheel @@ -1269,7 +1269,7 @@ jobs: DESIRED_CUDA: cu117 GPU_ARCH_VERSION: 11.7 GPU_ARCH_TYPE: cuda - DOCKER_IMAGE: pytorch/manylinux-builder:cuda11.7 + DOCKER_IMAGE: tousif111/manylinux-builder:cuda11.7 DESIRED_PYTHON: "3.9" build_name: manywheel-py3_9-cuda11_7-with-pypi-cudnn secrets: @@ -1290,7 +1290,7 @@ jobs: DESIRED_CUDA: cu117 GPU_ARCH_VERSION: 11.7 GPU_ARCH_TYPE: cuda - DOCKER_IMAGE: pytorch/manylinux-builder:cuda11.7 + DOCKER_IMAGE: tousif111/manylinux-builder:cuda11.7 DESIRED_PYTHON: "3.9" build_name: manywheel-py3_9-cuda11_7 build_environment: linux-binary-manywheel @@ -1310,7 +1310,7 @@ jobs: DESIRED_CUDA: cu117 GPU_ARCH_VERSION: 11.7 GPU_ARCH_TYPE: cuda - DOCKER_IMAGE: pytorch/manylinux-builder:cuda11.7 + DOCKER_IMAGE: tousif111/manylinux-builder:cuda11.7 DESIRED_PYTHON: "3.9" build_name: manywheel-py3_9-cuda11_7 build_environment: linux-binary-manywheel @@ -1329,7 +1329,7 @@ jobs: DESIRED_CUDA: cu117 GPU_ARCH_VERSION: 11.7 GPU_ARCH_TYPE: cuda - DOCKER_IMAGE: pytorch/manylinux-builder:cuda11.7 + DOCKER_IMAGE: tousif111/manylinux-builder:cuda11.7 DESIRED_PYTHON: "3.9" build_name: manywheel-py3_9-cuda11_7 secrets: @@ -1761,11 +1761,11 @@ jobs: DESIRED_CUDA: cu117 GPU_ARCH_VERSION: 11.7 GPU_ARCH_TYPE: cuda - DOCKER_IMAGE: pytorch/manylinux-builder:cuda11.7 + DOCKER_IMAGE: tousif111/manylinux-builder:cuda11.7 DESIRED_PYTHON: "3.10" build_name: manywheel-py3_10-cuda11_7-with-pypi-cudnn build_environment: linux-binary-manywheel - PYTORCH_EXTRA_INSTALL_REQUIREMENTS: nvidia-cuda-runtime-cu11; platform_system == 'Linux' | nvidia-cuda-cupti-cu11==11.7.101; platform_system == 'Linux' | nvidia-cudnn-cu11==8.5.0.96; platform_system == 'Linux' | nvidia-cublas-cu11==11.10.3.66; platform_system == 'Linux' | nvidia-cufft-cu11==10.9.0.58; platform_system == 'Linux' | nvidia-curand-cu11==10.2.10.91; platform_system == 'Linux' | nvidia-cusolver-cu11==11.4.0.1; platform_system == 'Linux' | nvidia-cusparse-cu11==11.7.4.91; platform_system == 'Linux' | nvidia-nvtx-cu11==11.7.91; platform_system == 'Linux' + PYTORCH_EXTRA_INSTALL_REQUIREMENTS: nvidia-cuda-nvrtc-cu11==11.7.99; platform_system == 'Linux' | nvidia-cuda-runtime-cu11==11.7.99; platform_system == 'Linux' | nvidia-cuda-cupti-cu11==11.7.101; platform_system == 'Linux' | nvidia-cudnn-cu11==8.5.0.96; platform_system == 'Linux' | nvidia-cublas-cu11==11.10.3.66; platform_system == 'Linux' | nvidia-cufft-cu11==10.9.0.58; platform_system == 'Linux' | nvidia-curand-cu11==10.2.10.91; platform_system == 'Linux' | nvidia-cusolver-cu11==11.4.0.1; platform_system == 'Linux' | nvidia-cusparse-cu11==11.7.4.91; platform_system == 'Linux' | nvidia-nccl-cu11==2.14.3; platform_system == 'Linux' | nvidia-nvtx-cu11==11.7.91; platform_system == 'Linux' secrets: github-token: ${{ secrets.GITHUB_TOKEN }} @@ -1782,7 +1782,7 @@ jobs: DESIRED_CUDA: cu117 GPU_ARCH_VERSION: 11.7 GPU_ARCH_TYPE: cuda - DOCKER_IMAGE: pytorch/manylinux-builder:cuda11.7 + DOCKER_IMAGE: tousif111/manylinux-builder:cuda11.7 DESIRED_PYTHON: "3.10" build_name: manywheel-py3_10-cuda11_7-with-pypi-cudnn build_environment: linux-binary-manywheel @@ -1801,7 +1801,7 @@ jobs: DESIRED_CUDA: cu117 GPU_ARCH_VERSION: 11.7 GPU_ARCH_TYPE: cuda - DOCKER_IMAGE: pytorch/manylinux-builder:cuda11.7 + DOCKER_IMAGE: tousif111/manylinux-builder:cuda11.7 DESIRED_PYTHON: "3.10" build_name: manywheel-py3_10-cuda11_7-with-pypi-cudnn secrets: @@ -1822,7 +1822,7 @@ jobs: DESIRED_CUDA: cu117 GPU_ARCH_VERSION: 11.7 GPU_ARCH_TYPE: cuda - DOCKER_IMAGE: pytorch/manylinux-builder:cuda11.7 + DOCKER_IMAGE: tousif111/manylinux-builder:cuda11.7 DESIRED_PYTHON: "3.10" build_name: manywheel-py3_10-cuda11_7 build_environment: linux-binary-manywheel @@ -1842,7 +1842,7 @@ jobs: DESIRED_CUDA: cu117 GPU_ARCH_VERSION: 11.7 GPU_ARCH_TYPE: cuda - DOCKER_IMAGE: pytorch/manylinux-builder:cuda11.7 + DOCKER_IMAGE: tousif111/manylinux-builder:cuda11.7 DESIRED_PYTHON: "3.10" build_name: manywheel-py3_10-cuda11_7 build_environment: linux-binary-manywheel @@ -1861,7 +1861,7 @@ jobs: DESIRED_CUDA: cu117 GPU_ARCH_VERSION: 11.7 GPU_ARCH_TYPE: cuda - DOCKER_IMAGE: pytorch/manylinux-builder:cuda11.7 + DOCKER_IMAGE: tousif111/manylinux-builder:cuda11.7 DESIRED_PYTHON: "3.10" build_name: manywheel-py3_10-cuda11_7 secrets: @@ -2293,11 +2293,11 @@ jobs: DESIRED_CUDA: cu117 GPU_ARCH_VERSION: 11.7 GPU_ARCH_TYPE: cuda - DOCKER_IMAGE: pytorch/manylinux-builder:cuda11.7 + DOCKER_IMAGE: tousif111/manylinux-builder:cuda11.7 DESIRED_PYTHON: "3.11" build_name: manywheel-py3_11-cuda11_7-with-pypi-cudnn build_environment: linux-binary-manywheel - PYTORCH_EXTRA_INSTALL_REQUIREMENTS: nvidia-cuda-runtime-cu11; platform_system == 'Linux' | nvidia-cuda-cupti-cu11==11.7.101; platform_system == 'Linux' | nvidia-cudnn-cu11==8.5.0.96; platform_system == 'Linux' | nvidia-cublas-cu11==11.10.3.66; platform_system == 'Linux' | nvidia-cufft-cu11==10.9.0.58; platform_system == 'Linux' | nvidia-curand-cu11==10.2.10.91; platform_system == 'Linux' | nvidia-cusolver-cu11==11.4.0.1; platform_system == 'Linux' | nvidia-cusparse-cu11==11.7.4.91; platform_system == 'Linux' | nvidia-nvtx-cu11==11.7.91; platform_system == 'Linux' + PYTORCH_EXTRA_INSTALL_REQUIREMENTS: nvidia-cuda-nvrtc-cu11==11.7.99; platform_system == 'Linux' | nvidia-cuda-runtime-cu11==11.7.99; platform_system == 'Linux' | nvidia-cuda-cupti-cu11==11.7.101; platform_system == 'Linux' | nvidia-cudnn-cu11==8.5.0.96; platform_system == 'Linux' | nvidia-cublas-cu11==11.10.3.66; platform_system == 'Linux' | nvidia-cufft-cu11==10.9.0.58; platform_system == 'Linux' | nvidia-curand-cu11==10.2.10.91; platform_system == 'Linux' | nvidia-cusolver-cu11==11.4.0.1; platform_system == 'Linux' | nvidia-cusparse-cu11==11.7.4.91; platform_system == 'Linux' | nvidia-nccl-cu11==2.14.3; platform_system == 'Linux' | nvidia-nvtx-cu11==11.7.91; platform_system == 'Linux' secrets: github-token: ${{ secrets.GITHUB_TOKEN }} @@ -2314,7 +2314,7 @@ jobs: DESIRED_CUDA: cu117 GPU_ARCH_VERSION: 11.7 GPU_ARCH_TYPE: cuda - DOCKER_IMAGE: pytorch/manylinux-builder:cuda11.7 + DOCKER_IMAGE: tousif111/manylinux-builder:cuda11.7 DESIRED_PYTHON: "3.11" build_name: manywheel-py3_11-cuda11_7-with-pypi-cudnn build_environment: linux-binary-manywheel @@ -2333,7 +2333,7 @@ jobs: DESIRED_CUDA: cu117 GPU_ARCH_VERSION: 11.7 GPU_ARCH_TYPE: cuda - DOCKER_IMAGE: pytorch/manylinux-builder:cuda11.7 + DOCKER_IMAGE: tousif111/manylinux-builder:cuda11.7 DESIRED_PYTHON: "3.11" build_name: manywheel-py3_11-cuda11_7-with-pypi-cudnn secrets: @@ -2354,7 +2354,7 @@ jobs: DESIRED_CUDA: cu117 GPU_ARCH_VERSION: 11.7 GPU_ARCH_TYPE: cuda - DOCKER_IMAGE: pytorch/manylinux-builder:cuda11.7 + DOCKER_IMAGE: tousif111/manylinux-builder:cuda11.7 DESIRED_PYTHON: "3.11" build_name: manywheel-py3_11-cuda11_7 build_environment: linux-binary-manywheel @@ -2374,7 +2374,7 @@ jobs: DESIRED_CUDA: cu117 GPU_ARCH_VERSION: 11.7 GPU_ARCH_TYPE: cuda - DOCKER_IMAGE: pytorch/manylinux-builder:cuda11.7 + DOCKER_IMAGE: tousif111/manylinux-builder:cuda11.7 DESIRED_PYTHON: "3.11" build_name: manywheel-py3_11-cuda11_7 build_environment: linux-binary-manywheel @@ -2393,7 +2393,7 @@ jobs: DESIRED_CUDA: cu117 GPU_ARCH_VERSION: 11.7 GPU_ARCH_TYPE: cuda - DOCKER_IMAGE: pytorch/manylinux-builder:cuda11.7 + DOCKER_IMAGE: tousif111/manylinux-builder:cuda11.7 DESIRED_PYTHON: "3.11" build_name: manywheel-py3_11-cuda11_7 secrets: