fix base image

allenai · Nov 5, 2021 · 2897d1d · 2897d1d
1 parent ca7e79a
commit 2897d1d
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/.github/workflows/ci.yml b/.github/workflows/ci.yml
@@ -554,7 +554,7 @@ jobs:
     runs-on: ubuntu-latest
     strategy:
       matrix:
-        cuda: ['10.2', '11.1']
+        cuda: ['11.3']
 
     steps:
     - uses: actions/checkout@v2
@@ -597,7 +597,7 @@ jobs:
     - name: Upload latest image
       # HACK: we don't publish images for releases in the core library at the moment so this step would fail.
       # The CUDA condition is because CUDA 10.2 is currently our default.
-      if: github.event_name == 'release' && matrix.cuda == '10.2'
+      if: github.event_name == 'release' && matrix.cuda == '11.3'
       run: |
         docker tag $DOCKER_IMAGE_NAME allennlp/models:latest
         docker push allennlp/models:latest