Skip to content

Commit

Permalink
Adds more nvidia pypi dependencies (pytorch#89944)
Browse files Browse the repository at this point in the history
This PR adds more nvidia pypi dependencies for cuda 11.7 wheel. Additionally, it pins cufft version to 10.9.0.58 to resolve pytorch#88038

Depends on: pytorch/builder#1196

Pull Request resolved: pytorch#89944
Approved by: https://github.com/atalman
  • Loading branch information
syed-ahmed authored and razarmehr committed Jan 4, 2023
1 parent 4343de0 commit a9771c2
Show file tree
Hide file tree
Showing 2 changed files with 15 additions and 7 deletions.
12 changes: 10 additions & 2 deletions .github/scripts/generate_binary_build_matrix.py
Original file line number Diff line number Diff line change
Expand Up @@ -219,9 +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-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'",
"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'",
"build_name":
f"{package_type}-py{python_version}-{gpu_arch_type}{gpu_arch_version}-with-pypi-cudnn"
.replace(
Expand Down
10 changes: 5 additions & 5 deletions .github/workflows/generated-linux-binary-manywheel-nightly.yml

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

0 comments on commit a9771c2

Please sign in to comment.