Skip to content

remove index-url for 2.7.0 default pip wheel #1998

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Closed
wants to merge 1 commit into from

Conversation

ptrblck
Copy link
Contributor

@ptrblck ptrblck commented Apr 25, 2025

The current install matrix shows the --index-url for all CUDA versions for our latest PyTorch 2.7.0 release:
image

However, running pip install torch will install the default config: torch==2.7.0+cu126:

pip install torch
...
Collecting nvidia-cuda-nvrtc-cu12==12.6.77 (from torch)
  Using cached nvidia_cuda_nvrtc_cu12-12.6.77-py3-none-manylinux2014_x86_64.whl.metadata (1.5 kB)
Collecting nvidia-cuda-runtime-cu12==12.6.77 (from torch)
  Using cached nvidia_cuda_runtime_cu12-12.6.77-py3-none-manylinux2014_x86_64.manylinux_2_17_x86_64.whl.metadata (1.5 kB)
Collecting nvidia-cuda-cupti-cu12==12.6.80 (from torch)
  Using cached nvidia_cuda_cupti_cu12-12.6.80-py3-none-manylinux2014_x86_64.manylinux_2_17_x86_64.whl.metadata (1.6 kB)
...
Successfully installed nvidia-cublas-cu12-12.6.4.1 nvidia-cuda-cupti-cu12-12.6.80 nvidia-cuda-nvrtc-cu12-12.6.77 nvidia-cuda-runtime-cu12-12.6.77 nvidia-cudnn-cu12-9.5.1.17 nvidia-cufft-cu12-11.3.0.4 nvidia-cufile-cu12-1.11.1.6 nvidia-curand-cu12-10.3.7.77 nvidia-cusolver-cu12-11.7.1.2 nvidia-cusparse-cu12-12.5.4.2 nvidia-nvjitlink-cu12-12.6.85 nvidia-nvtx-cu12-12.6.77 torch-2.7.0

We should thus remove the unnecessary --index-url for 2.7.0+cu126.

CC @atalman @malfet

Copy link

netlify bot commented Apr 25, 2025

Deploy Preview for pytorch-dot-org-preview ready!

Name Link
🔨 Latest commit 489d72c
🔍 Latest deploy log https://app.netlify.com/sites/pytorch-dot-org-preview/deploys/680b9a7863afbf00085012f3
😎 Deploy Preview https://deploy-preview-1998--pytorch-dot-org-preview.netlify.app
📱 Preview on mobile
Toggle QR Code...

QR Code

Use your smartphone camera to open QR code link.

To edit notification comments on pull requests, go to your Netlify site configuration.

@atalman
Copy link
Contributor

atalman commented Apr 25, 2025

@ptrblck this pr is automated, I believe here is the automated one: #1996

@atalman
Copy link
Contributor

atalman commented Apr 25, 2025

I believe it was caused by this hardcoded value, which was fixed in this PR: pytorch/test-infra#6558

@atalman
Copy link
Contributor

atalman commented Apr 25, 2025

closing this one in favor of automated PR

@atalman atalman closed this Apr 25, 2025
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

Successfully merging this pull request may close these issues.

2 participants