Solved by ptrblck in post #2 
              
              
                Just get rid of all .cuda() calls and run the script. This prevents moving the model or tensors onto a GPU. 
Often there are optional arguments for the main script where you can specify using a GPU or not. 
EDIT: This code uses an argument for GPU usage. (Haven’t tried it myself)