Solved by Alexey_Demyanchuk in post #4 
              
              
                Hi) Most likely it is the torchvision version is different on your PC versus server. The opportunity to use transforms on tensors was added only recently. So pytorch and torchvision update on the server should help.