Hacktoberfest 2026: le issue che i maintainer hanno segnato per ottobre, aperte e adatte ai principianti. Sfoglia le issue Hacktoberfest

torchrec tutorial: old cuda version install

Aperta
  • Qualcosa non va?
#3,713 0 commenti 0 reazioni 0 assegnatari Vedi su GitHub

I maintainer di solito rispondono entro 2 giorni

Nessuno ha ancora preso questa issue.

Valutazione

Difficoltà
2/5
Tempo stimato
1-3 ore
Idoneità per principianti
48/100
Tipo di issue
Bug
Chiarezza
Specificata chiaramente
Stato di attività
Ferma
Stack tecnologico
python, pytorch
Ambito
documentation, machine-learning

Direzione di ricerca

Inizia da intermediate_source/torchrec_intro_tutorial.py alle righe 38-43 e riproduci l’installazione nell’ambiente Colab corrente. Verifica se le modifiche proposte al pacchetto e all’indice installano dipendenze torchrec compatibili, quindi esegui il tutorial abbastanza a lungo da confermare che la configurazione venga completata senza la mancata corrispondenza della versione di CUDA.

Scritto dal modello di indicizzazione a partire dal testo della issue.

Descrizione

torchrec

https://github.com/pytorch/tutorials/blob/a3695df13baac449f92a06b1029260a21f2e5c90/intermediate_source/torchrec_intro_tutorial.py#L38C1-L43C3

the current lines try to install a cuda 12.1 version when the current colab is 12.6 (and so torch remains at 12.6)
changing the index to cu126 fails, because torchrec requires tensordict, which is not in the index?

my proposed solution is to remove the pytorch install line
and add tensordict to torchmetrics line

# !pip3 install --pre torch --index-url https://download.pytorch.org/whl/cu121 -U remove
!pip3 install fbgemm_gpu --index-url https://download.pytorch.org/whl/cu126
!pip3 install torchmetrics==1.0.3 tensordict
!pip3 install torchrec --index-url https://download.pytorch.org/whl/cu126

cc @PaulZhang12 @iamzainhuda @vmoens @nairbv

Lingua principale
Python
Stelle
9.4k
Fork
4.4k
Merge medio
3g 5h
PR unite (30g)
4

Preparare l'ambiente

Apri in Codespaces

Avvia il container di sviluppo del progetto nel browser, con il tuo account GitHub.

Come iniziare

  1. Leggi tutta la issue e poi la guida ai contributi del progetto.
  2. Commenta sulla issue per dire che te ne occupi tu — evita che due persone facciano lo stesso lavoro.
  3. Fai un fork del repository e lavora su un branch.
  4. Apri una pull request che faccia riferimento al numero della issue.

Altre issue di pytorch/tutorials

Tutte le issue di pytorch/tutorials

Issue simili

Altre issue su Python

Ricevi le nuove issue nella tua casella

Un breve riepilogo di issue GitHub adatte ai principianti.