Error when running on Colab CPU instance about pytorch_memlab HOT 8 CLOSED

from pytorch_memlab import MemReporter

---------------------------------------------------------------------------

RuntimeError                              Traceback (most recent call last)

<ipython-input-4-f9afd24a6183> in <module>()
----> 1 from pytorch_memlab import MemReporter

5 frames

/usr/local/lib/python3.7/dist-packages/pytorch_memlab/__init__.py in <module>()
      1 from .courtesy import Courtesy
      2 from .mem_reporter import MemReporter
----> 3 from .line_profiler import LineProfiler, profile, profile_every, set_target_gpu, clear_global_line_profiler
      4 try:
      5     from .line_profiler.extension import load_ipython_extension

/usr/local/lib/python3.7/dist-packages/pytorch_memlab/line_profiler/__init__.py in <module>()
      1 from .line_profiler import LineProfiler
----> 2 from .profile import profile, profile_every, set_target_gpu, clear_global_line_profiler

/usr/local/lib/python3.7/dist-packages/pytorch_memlab/line_profiler/profile.py in <module>()
      4 
      5 global_line_profiler = LineProfiler()
----> 6 global_line_profiler.enable()
      7 
      8 

/usr/local/lib/python3.7/dist-packages/pytorch_memlab/line_profiler/line_profiler.py in enable(self)
     88         try:
     89             torch.cuda.empty_cache()
---> 90             self._reset_cuda_stats()
     91         # Pytorch-1.7.0 raises AttributeError while <1.6.0 raises AssertionError
     92         except (AssertionError, AttributeError) as error:

/usr/local/lib/python3.7/dist-packages/pytorch_memlab/line_profiler/line_profiler.py in _reset_cuda_stats(self)
     80 
     81     def _reset_cuda_stats(self):
---> 82         torch.cuda.reset_peak_memory_stats()
     83         torch.cuda.reset_accumulated_memory_stats()
     84 

/usr/local/lib/python3.7/dist-packages/torch/cuda/memory.py in reset_peak_memory_stats(device)
    236     """
    237     device = _get_device_index(device, optional=True)
--> 238     return torch._C._cuda_resetPeakMemoryStats(device)
    239 
    240 

RuntimeError: invalid argument to reset_peak_memory_stats