Problem with PyTorch

@Mo_O Please can you

  • post error messages from the screenshots as text
  • the output of nvidia-smi (as text)
  • the outcome a 3DFlex job, as suggested by @kookjookeem