PyTorch Nightly版を“爆速”インストールする最短ルート

概要

RTX 5060tiのようなRTX50シリーズは、普通のpytorchに対応していないようなので、pytorch nightly版をインストールしてみました。
（ほかの記事でも紹介されていますが、私は他の方の記事のやり方でうまくインストールできなかったので、私なりのやり方を紹介します。）

インストール方法

pip install torch==2.9.0.dev20250716 torchvision==0.24.0.dev20250717 torchaudio==2.8.0.dev20250717 --index-url https://download.pytorch.org/whl/nightly/cu129

解説

以下を実行すると、リリース時期が5月から7月までのnightly版をインストールし始め、なかなかpytorchのインストールが完了しませんでした。

pip install --pre torch torchvision torchaudio --extra-index-url https://download.pytorch.org/whl/nightly/cu129

そこでインストール方法に記載した内容で、直接バージョンを指定することで、インストール時間を短縮させました。
この方法は、指定したバージョンのみインストールします。

• URL末尾のcu129とは、CUDAのバージョンが12.9ということを意味しています。
• torchとtorchvision/torchaudioのリリース日が異なるのは、依存関係が原因です。
• 基本的にはtorch, torchvision, torchaudioの依存関係は同じリリース日が対応しているようですが、異なる日付が対応している場合もあります。
• まず同じリリース日のものを指定してインストールを試す。それでエラーが出たら、エラー文に記載されているバージョンを指定して再度インストールする。というのが良いかもしれません。

最後に

より良いインストール方法をご存じの方がいらっしゃれば、ご教授いただきたく思います。