search

mberr / torch-max-mem

Decorators for maximizing memory utilization with PyTorch & CUDA
https://torch-max-mem.readthedocs.io/en/latest/
MIT License
14 stars 0 forks source link

Import error #1

Closed mberr closed 2 years ago

mberr commented 2 years ago

When trying to run the example from the README, I currently get the following error

Traceback (most recent call last):
  File ".../torch_max_mem/tmp.py", line 2, in <module>
    from torch_max_mem import maximize_memory_utilization
ModuleNotFoundError: No module named 'torch_max_mem'

When I check pip list, the package name appears to be the stylized name

$ pip list | grep max
torch-max-mem     0.0.1.dev0 .../torch_max_mem/src
cthoyt commented 2 years ago

In b3470ff I added a unit test which seems to run alright (though the test currently fails ;) , maybe there's something funny with the way your installation/environment is working

mberr commented 2 years ago

I was in the wrong Python venv :sweat_smile: