diff --git a/.github/workflows/test-with-pytorch.yml b/.github/workflows/test-with-pytorch.yml index 031b79e..8e91a5e 100644 --- a/.github/workflows/test-with-pytorch.yml +++ b/.github/workflows/test-with-pytorch.yml @@ -31,6 +31,7 @@ jobs: 1.9.0-cuda10.2-cudnn7-runtime, 1.10.0-cuda11.3-cudnn8-runtime, 1.13.1-cuda11.6-cudnn8-runtime, + 2.1.2-cuda12.1-cudnn8-runtime, ] env: DOCKER_REPO: test-with-pytorch-${{matrix.pytorch-tag}}