Solved by KFrank in post #2 
              
              
                Hello Johan! 

I went through a similar issue, and was helped greatly by the answers I 
got in this thread (thanks, in particular, to @ptrblck and @peterjc123): 


In particular, one of these pytorch legacy builds worked for me. 
Note, going this route pushed me back to pytorch version 0.3.0. 
I sup…