Hacktoberfest 2026:メンテナが10月に向けて印を付けた、オープンで初心者向けの issue。 Hacktoberfest の issue を見る

torchrec tutorial: old cuda version install

オープン
#3,713 コメント 0 件 リアクション 0 件 担当者 0 名 GitHub で見る

メンテナーはふだん 2 日以内に返信

まだ誰も着手していません。

評価

難易度
2/5
見積もり時間
1〜3時間
初心者へのやさしさ
48/100
issue の種類
バグ
明瞭さ
明確に書かれている
活発さ
停滞
技術スタック
python, pytorch

調査の方向性

intermediate_source/torchrec_intro_tutorial.py の38-43行目から始め、現在のColab環境でインストールを再現します。提案されたパッケージとインデックスの変更によって互換性のあるtorchrec依存関係がインストールされるか確認し、CUDAバージョンの不一致なしにセットアップが完了することを確認できるところまでチュートリアルを実行します。

索引モデルが issue の本文から書いたものです。

説明

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

主要言語
Python
スター
9.4k
フォーク
4.4k
平均マージ
3日 5時間
マージ済み PR(30日)
4

環境構築

Codespaces で開く

このプロジェクトの開発コンテナを、あなたの GitHub アカウントでブラウザ上に起動します。

はじめの一歩

  1. issue を最後まで読み、次にプロジェクトのコントリビューションガイドを読みます。
  2. 着手することを issue にコメントします — 二人が同じ作業をするのを防げます。
  3. リポジトリをフォークし、ブランチを切って変更します。
  4. issue 番号を参照したプルリクエストを送ります。

pytorch/tutorials のほかの issue

pytorch/tutorials の issue をすべて見る

似ている issue

Python の issue をもっと見る

新しい issue をメールで受け取る

初心者向けの GitHub issue を短くまとめたダイジェスト。