torchrec tutorial: old cuda version install

Offen
  • Stimmt etwas nicht?
#3,713 0 Kommentare 0 Reaktionen 0 zugewiesene Personen Auf GitHub ansehen

Maintainer antworten meist innerhalb von 1 Tag

Dieses Issue hat noch niemand übernommen.

Bewertung

Schwierigkeit
2/5
Geschätzter Aufwand
1-3 Stunden
Anfängerfreundlichkeit
48/100
Issue-Typ
Bug
Klarheit
Klar beschrieben
Aktivitätsstatus
Veraltet
Tech-Stack
python, pytorch
Bereich
documentation, machine-learning

Rechercherichtung

Beginne mit intermediate_source/torchrec_intro_tutorial.py in den Zeilen 38-43 und reproduziere die Installation in der aktuellen Colab-Umgebung. Überprüfe, ob die vorgeschlagenen Änderungen an Paket und Index kompatible torchrec-Abhängigkeiten installieren, und führe das Tutorial weit genug aus, um zu bestätigen, dass die Einrichtung ohne den CUDA-Versionskonflikt abgeschlossen wird.

Vom Indexierungsmodell aus dem Issue-Text verfasst.

Beschreibung

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

Vorherrschende Sprache
Python
Sterne
9.4k
Forks
4.4k
Ø Merge
22 Std. 59 Min.
Gemergte PRs (30 T.)
3

Entwicklungsumgebung

In Codespaces öffnen

Startet den Dev-Container des Projekts im Browser, mit Ihrem eigenen GitHub-Konto.

Erste Schritte

  1. Lesen Sie das ganze Issue und danach den Beitragsleitfaden des Projekts.
  2. Schreiben Sie ins Issue, dass Sie es übernehmen — das erspart doppelte Arbeit.
  3. Forken Sie das Repository und arbeiten Sie in einem Branch.
  4. Öffnen Sie einen Pull Request, der die Issue-Nummer nennt.

Mehr aus pytorch/tutorials

Alle Issues in pytorch/tutorials

Ähnliche Issues

Weitere Issues zu Python

Neue Issues direkt in Ihr Postfach

Eine kurze Übersicht über anfängerfreundliche GitHub-Issues.