Yea this issue was discovered a few days ago.
The fix hasn’t been merged yet, but I pulled the commits and it worked for me!

  
      github.com/pytorch/pytorch
  
  
      
    
  


  Fix BatchNorm double backwards memory leak (v.0.2.0) 38



  by gchanan
  on 10:00PM - 07 Aug 17



  3 commits
  changed 4 files
  with 51 additions
  and 49 deletions.