[jastanton]

I tried that as well but resulted in an error:

AttributeError: module 'torch._C' has no attribute '_cuda_resetPeakMemoryStats'

https://gist.github.com/JAStanton/73673d249927588c93ee530d08...

[gonzzz]

hi jastanto. Im on an intel mac running into the same problem. Did you find a workaround?