rye で pytorch を追加/インストールする方法

なにもかもわからず、３時間ぐらい試行錯誤したので、忘れたときのために。

結論

rye init したあとに

1. pyproject.toml に sources を追加 (CPU版の場合)

[[tool.rye.sources]]
name = "torch"
url = "https://download.pytorch.org/whl/cpu"
type = "index"

※ CPU版以外がほしい場合は pytoach 公式サイトの Getting Started で適切なやつを選んで、pip3 install のオプションに指定されてるURLを指定する。

2. rye add の書き方

$ rye add "torch==2.0.1"    

で追加。"2.0.1+cpu" とかじゃなくてok

$ rye sync

でインストール。