🐉
WindowsのuvでCUDA 12.4のPyTorchをインストールする
わけあってWindowsでPyTorchの環境を構築する必要があり、せっかくなのでuvで管理したいと思い立ちました。
ただ、公式サイトのWindows欄にあるコマンドをuv用に変更して実行してみるも…
uv add torch torchvision --index-url https://download.pytorch.org/whl/cu124
error: Distribution `torch==2.5.0 @ registry+https://download.pytorch.org/whl/cu124` can't be installed because it doesn't have a source distribution or wheel for the current platform
なんか怒られちゃいました😢
一応
uv pip install torch torchvision --index-url https://download.pytorch.org/whl/cu124
とするとインストールできるんですが、せっかくのuvのメリットがなくなってしまうのでどうにかuvで管理できないかと調べてみました。
すると一日前に投稿されてた、この記事がめちゃくちゃ参考になりました！
ありがとうございます(._.)
上記の記事を参考にして、最終的に完成した pyproject.toml がこれです。
記事とは違い素直にCUDA版のPyTorchを入れてくるだけです。
[project]
name = "uv-cuda-test"
version = "0.1.0"
description = "Add your description here"
readme = "README.md"
requires-python = ">=3.12"
dependencies = [
"torch==2.5.0+cu124",
"torchvision==0.20.0+cu124",
]
[tool.uv.sources]
torch = { index = "torch-cuda" }
torchvision = { index = "torch-cuda" }
[[tool.uv.index]]
name = "torch-cuda"
url = "https://download.pytorch.org/whl/cu124"
explicit = true
これを手動で記入したあと、uv sync して、 uv run hello.py すると無事動きました！
Discussion