diff --git a/.github/workflows/unit_tests.yml b/.github/workflows/unit_tests.yml
index 9be04f070..f7df937b4 100644
--- a/.github/workflows/unit_tests.yml
+++ b/.github/workflows/unit_tests.yml
@@ -231,7 +231,7 @@ jobs:
   check-torch:
     runs-on: [ self-hosted, xeon5 ]
     container:
-      image: 10.0.13.31:5000/nvidia/cuda:130-ubuntu24.04_0415
+      image: 10.0.13.31:5000/nvidia/cuda:130-ubuntu24.04_0612
       options: --device /dev/dri --ipc=host --runtime=nvidia --gpus all
       volumes:
         - /monster/ci/env/entrypoint.sh:/entrypoint.sh