diff --git a/.github/workflows/unit_test.yml b/.github/workflows/unit_test.yml
index f2dcd26..3cfc7f3 100644
--- a/.github/workflows/unit_test.yml
+++ b/.github/workflows/unit_test.yml
@@ -58,8 +58,8 @@ jobs:
       matrix:
         container:
           - 'pytorch/pytorch:2.4.1-cuda12.1-cudnn9-runtime'
-          - 'pytorch/pytorch:2.2.2-cuda12.1-cudnn8-runtime'
-          - 'pytorch/pytorch:2.1.0-cuda11.8-cudnn8-runtime'
+          #- 'pytorch/pytorch:2.2.2-cuda12.1-cudnn8-runtime'
+          #- 'pytorch/pytorch:2.1.0-cuda11.8-cudnn8-runtime'
     container:
       image: ${{ matrix.container }}
     steps: