PyTorch + Poetryで不要なwheelがダウンロードされる

PoetryでPyTorchを管理しようとしたメモ。機械学習系は結局condaとかになるのかなぁ……？

TL;DR

わからん(2023/05/31)

GitHubにissueが立ってた

(前提) PyTorchのインストール方法（not Poetry）

Linux かつ CPU only の環境では以下（cpu onlyだと cuda がなくて動かないとかなるのでpip install torchは怪しい）

code:pytorch-install

pip3 install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cpu

Poetryで環境ごとにパッケージを変える方法

--index-urlは sourceの指定でできる

code:add-source

poetry source add --priority=explicit torch-cpu https://download.pytorch.org/whl/cpu

poetry add --source torch-cpu torch

platformの指定はplatform="linux"かmarkers="platform_system == \"Linux\""のノリでいける

platformがドキュメントに書かれていない（らしい。非推奨？）

markersのほうが細かい条件を指定できそう

markers="(platform_system != \"Darwin\" or platform_machine != \"arm64\") and (platform_system != \"Linux\" or platform_machine != \"aarch64\")"的な

code:pyproject.toml

torch = [

{version = "^2.0.1+cpu", source = "torch-cpu", platform = "linux"},

{version = "^2.0.1", platform = "darwin"}

]

tool.poetry.source

name = "torch-cpu"

priority = "explicit"

Poetryのバグ？

実行環境に関わらずwheelファイルを全部ダウンロードしてしまう

→pytorchだけで 3minくらい待たされて辛い

got this to workらしいけどどうなんだろう