fixes for issues 65 and 66 (#67)

Co-authored-by: John Calderon <john.calderon@centml.ai>
CentML · Oct 10, 2023 · 8a2153c · 8a2153c
1 parent 896d720
commit 8a2153c
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 2 deletions.
diff --git a/README.md b/README.md
@@ -25,6 +25,7 @@ To run DeepView.Profile, you need:
     The current PyTorch install supports CUDA capabilities sm_37 sm_50 sm_60 sm_70.
     If you want to use the NVIDIA GeForce RTX 3060 Ti GPU with PyTorch, please check the instructions at https://pytorch.org/get-started/locally/
     ```
+- **For new RTX 4000 GPUs you need to install pytorch with cuda11.8 [pip install torch==2.0.1+cu118 torchvision==0.15.2+cu118 torchaudio==2.0.2 --index-url https://download.pytorch.org/whl/cu118] otherwise CUPTI will not initialize correctly**
 
 ### Installation from PyPi
 

diff --git a/deepview_profile/profiler/utilization.py b/deepview_profile/profiler/utilization.py
@@ -158,7 +158,7 @@ def _populate_backward_data(self, node):
         for ch in node.children:
             self._populate_backward_data(ch)
 
-    @functools.cache
+    @functools.lru_cache(maxsize=None)
     def _can_match(self, f, b):
         if "aten" in f and "Backward0" in b:
             raw_f = f[len("aten::"):].lower().replace("_", "")

diff --git a/pyproject.toml b/pyproject.toml
@@ -28,7 +28,7 @@ python = "^3.7"
 pyyaml = "*"
 protobuf = "3.19.6"
 numpy = "^1.15.2"
-torch = "*"
+torch = "<=2.0.1"
 nvidia-ml-py3 = "*"
 toml = "^0.10.2"
 pyRAPL = "^0.2.3"