Solved by ptrblck in post #2 
              
              
                Try to load your state_dict or model with these arguments to force all tensors to be on CPU: 
torch.load('my_file.pt', map_location=lambda storage, loc: storage)