Macでも安心してuvでtorchなどをインストールする方法

概要

前回は、Macで深層学習のライブラリを使う際にも安心して開発するためのpoetryの構築方法をdockerでの環境構築方法について紹介しました。

https://zenn.dev/bamboo_nova/articles/acef07560c63d6

一方で、最近ではuvが流行ってきており、こちらの方がrust製なので高速に動作することが知られています。

しかし、uvでもmacのような環境下だとtorchなどの一部のライブラリのインストールでエラーが発生してしまいます。今回はuvでの深層学習向けの環境構築方法のTipsを紹介したいと思います。

結論

早速結論ですが、下記の設定を入れることが簡単に動作することができます。

[tool.uv]
required-environments = [
    "sys_platform == 'darwin' and platform_machine == 'x86_64'"
]

dockerのビルドに使う場合は、下記の用に事前に吐き出してpip install -r requirements.txtで終わりです。

uv export --format requirements-txt

poetryは結構面倒臭かった印象がありますが、uvは割と簡単にインストールできるので便利ですね。