forked from awwong1/torchprof
-
Notifications
You must be signed in to change notification settings - Fork 0
/
pyproject.toml
28 lines (25 loc) · 904 Bytes
/
pyproject.toml
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
[tool.poetry]
name = "torchprof"
version = "1.1.1"
description = "Profile PyTorch models"
license = "MIT"
classifiers = ["License :: OSI Approved :: MIT License", "Operating System :: OS Independent", "Programming Language :: Python :: 3", "Programming Language :: Python :: 3.5", "Programming Language :: Python :: 3.6", "Programming Language :: Python :: 3.7", "Programming Language :: Python :: 3.8"]
homepage = "https://github.com/indigoviolet/torchprof"
authors = ["Venky Iyer <[email protected]>"]
readme = "README.org"
[tool.poetry.dependencies]
python = ">=3.8,<4.0"
tqdm = ">=4.51.0"
tabulate = ">=0.8.7"
more-itertools = ">=8.6.0"
rich = ">=9.13.0"
varname = "^0.6.3"
[tool.poetry.dev-dependencies]
jupyter = ">=1.0.0"
mypy = ">=0.800"
torch = ">=1.7.1"
torchvision = ">=0.8.2"
snoop = "^0.3.0"
[build-system]
requires = ["poetry-core>=1.0.0"]
build-backend = "poetry.core.masonry.api"