49 lines
1.5 KiB
TOML
49 lines
1.5 KiB
TOML
[tool.poetry]
|
|
name = "reproducible"
|
|
version = "0.0.1"
|
|
description = "Reproducible Machine Learning Environment"
|
|
authors = ["Leni Aniva <v@leni.sh>"]
|
|
readme = "README.md"
|
|
include = []
|
|
packages = [{ include = "reproducible" }]
|
|
|
|
[tool.poetry.scripts]
|
|
reproducible = "reproducible.util:main"
|
|
|
|
[build-system]
|
|
requires = ["poetry-core"]
|
|
build-backend = "poetry.core.masonry.api"
|
|
|
|
[tool.poetry.dependencies]
|
|
python = "^3.11"
|
|
numpy = "^1.26"
|
|
#torch = { url = "https://download.pytorch.org/whl/cu118/torch-2.2.1+cu118-cp311-cp311-linux_x86_64.whl" }
|
|
torch = { version = "^2.2.0", source = "pytorch-cu118" }
|
|
torch-geometric = "^2.5.2"
|
|
#pyg-lib = { git = "https://github.com/pyg-team/pyg-lib.git", tag = "0.4.0" }
|
|
#torch-cluster = { version = "1.6.3" }
|
|
#torch-scatter = { version = "2.1.2" }
|
|
#torch-sparse = { version = "0.6.18" }
|
|
#torch-spline-conv = { version = "1.2.2" }
|
|
|
|
#pyg-lib = { version = "0.4.0", source = "pyg-torch220-cu118" }
|
|
#torch-cluster = { version = "1.6.3", source = "pyg-torch220-cu118" }
|
|
#torch-scatter = { version = "2.1.2", source = "pyg-torch220-cu118" }
|
|
#torch-sparse = { version = "0.6.18", source = "pyg-torch220-cu118" }
|
|
#torch-spline-conv = { version = "1.2.2", source = "pyg-torch220-cu118" }
|
|
|
|
[[tool.poetry.source]]
|
|
name = "pytorch-cu118"
|
|
url = "https://download.pytorch.org/whl/cu118"
|
|
priority = "explicit"
|
|
|
|
#[[tool.poetry.source]]
|
|
#name = "pyg-torch220-cu118"
|
|
#url = "git+https://github.com/pyg-team/pyg-lib.git"
|
|
#url = "https://data.pyg.org/whl/torch-2.2.0+cu118.html"
|
|
#priority = "explicit"
|
|
|
|
#[[tool.poetry.source]]
|
|
#name = "PyPI"
|
|
#priority = "primary"
|