Yes you can download the code file to your local directory.
Have u checked  if their’s any configuration that looks off on ur nnvidia panel and also have u tried updating ur gpu drivers
Also this is just a speculation, but I don’t think u should use the ‘+=’ operator in the training loop to update coz this can accumulate gradients
Can u find a better way for this.
Another thing I want to ask is this: does u neural network have only linear layers and no convolutional layers? If ur neural network has convolutional layers can u pls post your neural network code here pls let me see how it downsamples images
The thing is if u r using just linear layers it can cause memory issues for ur gpu and if u are convolutional layers that do not downsample these images it will cause similar issues to when u use only linear layers.