search

Stonesjtu / pytorch_memlab

Profiling and inspecting memory in pytorch
MIT License
1.02k stars 37 forks source link

ImportError: cannot import name 'set_target_gpu' #3

Closed ProGamerGov closed 5 years ago

ProGamerGov commented 5 years ago

I installed pytorch_memlab with pip3 install pytorch_memlab and I get this error when trying to set the target GPU:

    from pytorch_memlab import profile, set_target_gpu
ImportError: cannot import name 'set_target_gpu'
Stonesjtu commented 5 years ago

The newest package is not uploaded to pip yet. Can you plz try install from source via

pip install git+https://github.com/stonesjtu/pytorch_memlab
Stonesjtu commented 5 years ago

I just uploaded the current master to PyPI. you can try to update the package by pip install -U pytorch_memlab.

Can you please verify if the problem exists with the newest package?

Stonesjtu commented 5 years ago

Should be fixed, I'm closing this issue. Please feel free to re-open if the problem still exists with the newest release.