Giters

Stonesjtu / pytorch_memlab

Profiling and inspecting memory in pytorch

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

ImportError: cannot import name 'set_target_gpu'

ProGamerGov opened this issue · comments

commented

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'
commented

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
commented

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?

commented

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