diff --git a/.github/workflows/ci.yml b/.github/workflows/ci.yml index 8bf2082..267bd9d 100644 --- a/.github/workflows/ci.yml +++ b/.github/workflows/ci.yml @@ -40,7 +40,7 @@ jobs: include: - script: test coverage-name: gpu - tf-version: tensorflow[and-cuda] + tf-version: tensorflow[and-cuda]==2.16.1 runs-on: [self-hosted, gpu] - script: test python-version: "3.11"