v1.4.0

- updated README with deprecation notice
- cap pytorch support to 1.9

Author: awwong1

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.

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",

torchprof/__init__.py

@@ -3,5 +3,5 @@
 name = "torchprof"
 
 __all__ = ["Profile"]
-__version__ = "1.3.4"
+__version__ = "1.4.0"