Update CI GPU to run on CUDA 11.7 (#2137)

Co-authored-by: Mark Saroufim <marksaroufim@fb.com>
pytorch · Feb 16, 2023 · 8b3ae1e · 8b3ae1e
1 parent 7a7cc69
commit 8b3ae1e
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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: