diff --git a/.github/workflows/ci_gpu.yml b/.github/workflows/ci_gpu.yml index 17e55f1690..e43e2b92f2 100644 --- a/.github/workflows/ci_gpu.yml +++ b/.github/workflows/ci_gpu.yml @@ -38,7 +38,7 @@ jobs: uses: actions/checkout@v3 - name: Install dependencies run: | - python ts_scripts/install_dependencies.py --environment=dev --cuda=cu102 + python ts_scripts/install_dependencies.py --environment=dev --cuda=cu117 - name: Torchserve Sanity uses: nick-fields/retry@v2 with: