Solved by hgfedcba in post #4 
              
              
                I figured out the main problrm that was my installation. I hadn’t installed torch with cuda support and torch 1.4.1 didn’t send warnings about this. It said cuda was available which confused me, but I didn’t question it. 
After I updated the torch package version 1.7.1 triggered an assertion that I …