Skip to content

Commit

Permalink
Add platform markers for linux x86_64 only extra_install_requires (#9…
Browse files Browse the repository at this point in the history
…3066)

Like #89924 #91083

#85097 added new extra dependencies on nvidia-*. They are linux x86_64 (GPU) only packages, but were not marked as such, causing issues installing pytorch 1.13 via Poetry (and possibly other tools that follow PyPI's metadata API) on Linux aarch64 systems. This "fixes" the issue by adding the `and platform_machine == 'x86_64'` marker on these dependencies.

Pull Request resolved: #93066
Approved by: https://github.com/malfet
  • Loading branch information
TimeRainStarSky authored and pytorchmergebot committed Jan 31, 2023
1 parent 18c6ca1 commit b179a09
Show file tree
Hide file tree
Showing 2 changed files with 15 additions and 15 deletions.
22 changes: 11 additions & 11 deletions .github/scripts/generate_binary_build_matrix.py
Expand Up @@ -219,17 +219,17 @@ def generate_wheels_matrix(os: str,
"container_image": WHEEL_CONTAINER_IMAGES[arch_version],
"package_type": package_type,
"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'",
"nvidia-cuda-nvrtc-cu11==11.7.99; platform_system == 'Linux' and platform_machine == 'x86_64' | "
"nvidia-cuda-runtime-cu11==11.7.99; platform_system == 'Linux' and platform_machine == 'x86_64' | "
"nvidia-cuda-cupti-cu11==11.7.101; platform_system == 'Linux' and platform_machine == 'x86_64' | "
"nvidia-cudnn-cu11==8.5.0.96; platform_system == 'Linux' and platform_machine == 'x86_64' | "
"nvidia-cublas-cu11==11.10.3.66; platform_system == 'Linux' and platform_machine == 'x86_64' | "
"nvidia-cufft-cu11==10.9.0.58; platform_system == 'Linux' and platform_machine == 'x86_64' | "
"nvidia-curand-cu11==10.2.10.91; platform_system == 'Linux' and platform_machine == 'x86_64' | "
"nvidia-cusolver-cu11==11.4.0.1; platform_system == 'Linux' and platform_machine == 'x86_64' | "
"nvidia-cusparse-cu11==11.7.4.91; platform_system == 'Linux' and platform_machine == 'x86_64' | "
"nvidia-nccl-cu11==2.14.3; platform_system == 'Linux' and platform_machine == 'x86_64' | "
"nvidia-nvtx-cu11==11.7.91; platform_system == 'Linux' and platform_machine == 'x86_64'",
"build_name":
f"{package_type}-py{python_version}-{gpu_arch_type}{gpu_arch_version}-with-pypi-cudnn"
.replace(
Expand Down

Some generated files are not rendered by default. Learn more about how customized files appear on GitHub.

0 comments on commit b179a09

Please sign in to comment.