From 3ba23bdb8f00d04272f240c4724541be5fd09464 Mon Sep 17 00:00:00 2001 From: Dong Hyuk Chang Date: Wed, 7 Aug 2024 20:40:00 -0400 Subject: [PATCH] Update base image for tts asr import check test (#10072) * Fix torch version for tts asr import check test Signed-off-by: Dong Hyuk Chang * Ignore torch requirement Signed-off-by: Dong Hyuk Chang * Update base image used for import check Signed-off-by: Dong Hyuk Chang --------- Signed-off-by: Dong Hyuk Chang Co-authored-by: Dong Hyuk Chang --- .github/workflows/import-test.yml | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/.github/workflows/import-test.yml b/.github/workflows/import-test.yml index 6f2f52bfb0ae..3af15294b2a2 100644 --- 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 \ No newline at end of file + pip uninstall -y nemo_toolkit