opdx / pyproject.toml
lyangas
torch version
2e06513
raw
history blame contribute delete
672 Bytes
[tool.poetry]
name = "icd10_docker"
version = "0.1.0"
description = ""
authors = ["lyangasovid <[email protected]>"]
[tool.poetry.dependencies]
python = "^3.9"
scikit-learn = "1.3.0"
transformers = "4.29.2"
flask = "2.0.3"
xgboost = "1.7.6"
werkzeug = "2.2.2"
numpy = "1.26.3"
pandas = "2.2.0"
tqdm = "4.66.1"
torch = {version = "^2.3.1+cpu", source = "pytorch_cpu"}
torchvision = {version = "^0.18.1+cpu", source = "pytorch_cpu"}
[tool.poetry.dev-dependencies]
[[tool.poetry.source]]
name = "pytorch_cpu"
url = "https://download.pytorch.org/whl/cpu"
priority = "explicit"
[build-system]
requires = ["poetry-core>=1.0.0"]
build-backend = "poetry.core.masonry.api"