Update base image for tts asr import check test (NVIDIA#10072)

* Fix torch version for tts asr import check test Signed-off-by: Dong Hyuk Chang <[email protected]> * Ignore torch requirement Signed-off-by: Dong Hyuk Chang <[email protected]> * Update base image used for import check Signed-off-by: Dong Hyuk Chang <[email protected]> --------- Signed-off-by: Dong Hyuk Chang <[email protected]> Co-authored-by: Dong Hyuk Chang <[email protected]>
paarthneekhara · Jan 18, 2025 · 222aabd · 222aabd
1 parent 2b62977
commit 222aabd
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/.github/workflows/import-test.yml b/.github/workflows/import-test.yml
@@ -12,7 +12,7 @@ jobs:
   test-asr-imports:
     runs-on: ubuntu-latest
     container:
-      image: pytorch/pytorch:2.0.1-cuda11.7-cudnn8-runtime
+      image: pytorch/pytorch:2.4.0-cuda11.8-cudnn9-runtime
     steps:
     - name: Checkout repo
       uses: actions/checkout@v2
@@ -43,7 +43,7 @@ jobs:
   test-tts-imports:
     runs-on: ubuntu-latest
     container:
-      image: pytorch/pytorch:2.0.1-cuda11.7-cudnn8-runtime
+      image: pytorch/pytorch:2.4.0-cuda11.8-cudnn9-runtime
     steps:
     - name: Checkout repo
       uses: actions/checkout@v2
@@ -70,4 +70,4 @@ jobs:
         # Run import checks
         python tests/core_ptl/check_imports.py --domain "tts"
         # Uninstall NeMo
-        pip uninstall -y nemo_toolkit
+        pip uninstall -y nemo_toolkit