Untitled

[tool.poetry]
name = "koila"
version = "0.2.0"
description = "Prevent PyTorch's `CUDA error out of memory` in a few lines of code."
authors = ["rentruewang <patrick1031wang@gmail.com>"]
license = "MIT"
repository = "https://github.com/rentruewang/koila"

[tool.poetry.dependencies]
numpy = "^1.22.0"
python = ">=3.10,<3.11"
pynvml = "^11.4.1"
rich = "^13.4.1"
torch = "^2.0.1"

[tool.poetry.group.dev.dependencies]
autoflake = "^2.1.1"
black = "^23.3.0"
coverage = "^7.2.7"
isort = "^5.12.0"
mkdocs = "^1.4.3"
mkdocs-git-revision-date-plugin = "^0.3.1"
mkdocs-git-revision-date-localized-plugin = "^0.11.1"
mkdocs-material = "^9.1.15"
mypy = "^1.8.0"
pytest = "^7.3.1"

[build-system]
requires = ["poetry-core"]
build-backend = "poetry.core.masonry.api"