20 lines
429 B
TOML
20 lines
429 B
TOML
|
[project]
|
||
|
|
name = "python-uv-torch-cu124"
|
||
|
|
version = "0.1.0"
|
||
|
|
description = "Add your description here"
|
||
|
|
readme = "README.md"
|
||
|
|
requires-python = ">=3.11"
|
||
|
|
dependencies = [
|
||
|
|
"torch==2.5.1+cu124",
|
||
|
|
"torchvision==0.20.1+cu124",
|
||
|
|
]
|
||
|
|
|
||
|
|
[tool.uv.sources]
|
||
|
|
torch = { index = "pytorch-cu124" }
|
||
|
|
torchvision = { index = "pytorch-cu124" }
|
||
|
|
|
||
|
|
[[tool.uv.index]]
|
||
|
|
name = "pytorch-cu124"
|
||
|
|
url = "https://download.pytorch.org/whl/cu124"
|
||
|
|
explicit = true
|