Skip to content

Commit 66b3325

Browse files
syed-ahmedpytorchmergebot
authored andcommitted
Adds more nvidia pypi dependencies (pytorch#89944)
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
1 parent e26cb06 commit 66b3325

File tree

2 files changed

+15
-7
lines changed

2 files changed

+15
-7
lines changed

.github/scripts/generate_binary_build_matrix.py

+10-2
Original file line numberDiff line numberDiff line change
@@ -219,9 +219,17 @@ def generate_wheels_matrix(os: str,
219219
"container_image": WHEEL_CONTAINER_IMAGES[arch_version],
220220
"package_type": package_type,
221221
"pytorch_extra_install_requirements":
222-
"nvidia-cuda-runtime-cu11; platform_system == 'Linux' | "
222+
"nvidia-cuda-nvrtc-cu11==11.7.99; platform_system == 'Linux' | "
223+
"nvidia-cuda-runtime-cu11==11.7.99; platform_system == 'Linux' | "
224+
"nvidia-cuda-cupti-cu11==11.7.101; platform_system == 'Linux' | "
223225
"nvidia-cudnn-cu11==8.5.0.96; platform_system == 'Linux' | "
224-
"nvidia-cublas-cu11==11.10.3.66; platform_system == 'Linux'",
226+
"nvidia-cublas-cu11==11.10.3.66; platform_system == 'Linux' | "
227+
"nvidia-cufft-cu11==10.9.0.58; platform_system == 'Linux' | "
228+
"nvidia-curand-cu11==10.2.10.91; platform_system == 'Linux' | "
229+
"nvidia-cusolver-cu11==11.4.0.1; platform_system == 'Linux' | "
230+
"nvidia-cusparse-cu11==11.7.4.91; platform_system == 'Linux' | "
231+
"nvidia-nccl-cu11==2.14.3; platform_system == 'Linux' | "
232+
"nvidia-nvtx-cu11==11.7.91; platform_system == 'Linux'",
225233
"build_name":
226234
f"{package_type}-py{python_version}-{gpu_arch_type}{gpu_arch_version}-with-pypi-cudnn"
227235
.replace(

.github/workflows/generated-linux-binary-manywheel-nightly.yml

+5-5
Some generated files are not rendered by default. Learn more about customizing how changed files appear on GitHub.

0 commit comments

Comments
 (0)