v1.4.0

- updated README with deprecation notice - cap pytorch support to 1.9
awwong1 · May 23, 2021 · adcb3ca · adcb3ca
1 parent ee1e25c
commit adcb3ca
Show file tree

Hide file tree

Showing 3 changed files with 4 additions and 2 deletions.
diff --git a/README.md b/README.md
@@ -3,6 +3,8 @@
 [![PyPI version](https://badge.fury.io/py/torchprof.svg)](https://pypi.org/project/torchprof/)
 [![CircleCI](https://circleci.com/gh/awwong1/torchprof.svg?style=svg)](https://circleci.com/gh/awwong1/torchprof)
 
+> Attention! [This library is deprecated due to the PyTorch 1.9 changes to the torch profiler. Please use the official profiler.](https://pytorch.org/docs/1.9.0/profiler.html?highlight=profiler#module-torch.profiler) Thank you!
+
 A minimal dependency library for layer-by-layer profiling of PyTorch models.
 
 All metrics are derived using the PyTorch autograd profiler.

diff --git a/setup.py b/setup.py
@@ -34,7 +34,7 @@ def find_version(*file_paths):
  url="https://github.com/awwong1/torchprof",
  packages=setuptools.find_packages(),
  license="MIT",
- install_requires=["torch>=1.1.0,<2"],
+ install_requires=["torch>=1.1.0,<1.9"],
  classifiers=[
  "License :: OSI Approved :: MIT License",
  "Operating System :: OS Independent",

diff --git a/torchprof/__init__.py b/torchprof/__init__.py
@@ -3,5 +3,5 @@
 name = "torchprof"
 
 __all__ = ["Profile"]
-__version__ = "1.3.4"
+__version__ = "1.4.0"