Solved by aribenjamin in post #13 
              
              
                It seems this issue has been solved in more recent releases. Check out the Github issue opened by the OP: (https://github.com/pytorch/pytorch/issues/1776 )