Switching to NGC PyTorch container

Author: akshaysubr

.github/workflows/ci.yaml

@@ -36,7 +36,7 @@ jobs:
 
     runs-on: ubuntu-latest
     container:
-      image: pytorch/pytorch:latest
+      image: nvcr.io/nvidia/pytorch:25.04-py3
 
     steps:
     - name: Checkout repository