Open1

自分用設定済み uv の pyproject.toml

PlatPlat

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