Open1
自分用設定済み uv の pyproject.toml
Python 3.11 + PyTorch 2.4(cu124) + flash-attn + Jupyter
[project]
name = お名前 # "my-package"
version = "0.1.0"
description = "Add your description here"
readme = "README.md"
requires-python = ">=3.11,<3.12"
dependencies = [
"torch<2.5,>=2.4",
"torchvision>=0.19.0",
"flash-attn>=2.7.2.post1",
]
[tool.uv]
dev-dependencies = [
"jupyter>=1.1.1",
"ruff>=0.8.3",
"ipykernel>=6.29.5",
"matplotlib>=3.9.4",
"pytest>=8.3.4",
]
no-build-isolation-package = ["flash-attn"]
[tool.uv.sources]
torch = [{ index = "pytorch-cu124", marker = "platform_system != 'Darwin'" }]
torchvision = [
{ index = "pytorch-cu124", marker = "platform_system != 'Darwin'" },
]
[[tool.uv.index]]
name = "pytorch-cu124"
url = "https://download.pytorch.org/whl/cu124"
explicit = true