Adds more nvidia pypi dependencies#89944
Conversation
🔗 Helpful Links🧪 See artifacts and rendered test results at hud.pytorch.org/pr/89944
Note: Links to docs will display an error until the docs builds have been completed. ✅ No FailuresAs of commit ed682c3: This comment was automatically generated by Dr. CI and updates every 15 minutes. |
|
@pytorchbot label ciflow/binaries_wheel |
|
Can't add following labels to PR: ciflow/binaries_wheel Please ping one of the reviewers for help. |
fbc0fdc to
0179561
Compare
There was a problem hiding this comment.
This PR looks good. However it needs to be merged togeather with builder change. Before merging all the temporary strings needs to be removed and replaced with correct ones.
Also we will need to run some smoke tests on this package before merging the changes.
0179561 to
e84b1f2
Compare
|
@pytorchbot merge |
|
This PR needs to be approved by an authorized maintainer before merge. |
|
@pytorchbot merge |
|
This PR needs to be approved by an authorized maintainer before merge. |
|
@pytorchbot merge -f "CI/CD is green" |
Merge startedYour change will be merged immediately since you used the force (-f) flag, bypassing any CI checks (ETA: 1-5 minutes). Learn more about merging in the wiki. Questions? Feedback? Please reach out to the PyTorch DevX Team |
This PR adds more nvidia pypi dependencies for cuda 11.7 wheel. Additionally, it pins cufft version to 10.9.0.58 to resolve #88038
Depends on: pytorch/builder#1196
cc: @malfet @atalman @ptrblck