diff --git a/.github/scripts/generate_binary_build_matrix.py b/.github/scripts/generate_binary_build_matrix.py index 70c69fde929fa0..a32cba2cfa7bd1 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 192ca251b79ff0..8f52e67530254a 100644 --- a/.github/workflows/_binary-build-linux.yml +++ b/.github/workflows/_binary-build-linux.yml @@ -154,12 +154,12 @@ jobs: git clean -fxd working-directory: pytorch - - name: Checkout pytorch/builder to builder dir + - name: Checkout syed-ahmed/builder to builder dir uses: zhouzhuojie/checkout@05b13c9a0d21f08f6d5e64a1d5042246d13619d9 with: ref: main submodules: recursive - repository: pytorch/builder + repository: syed-ahmed/builder path: builder - name: Clean pytorch/builder checkout run: | diff --git a/.github/workflows/generated-linux-binary-manywheel-nightly.yml b/.github/workflows/generated-linux-binary-manywheel-nightly.yml index 3c00ce873cef52..d95e0fdc431664 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: @@ -693,11 +693,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 }} @@ -714,7 +714,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 @@ -733,7 +733,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: @@ -754,7 +754,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 @@ -774,7 +774,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 @@ -793,7 +793,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: @@ -1221,11 +1221,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 }} @@ -1242,7 +1242,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 @@ -1261,7 +1261,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: @@ -1282,7 +1282,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 @@ -1302,7 +1302,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 @@ -1321,7 +1321,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: @@ -1749,11 +1749,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 }} @@ -1770,7 +1770,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 @@ -1789,7 +1789,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: @@ -1810,7 +1810,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 @@ -1830,7 +1830,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 @@ -1849,7 +1849,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: @@ -2277,11 +2277,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 }} @@ -2298,7 +2298,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 @@ -2317,7 +2317,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: @@ -2338,7 +2338,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 @@ -2358,7 +2358,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 @@ -2377,7 +2377,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: