Skip to content

Commit

Permalink
Merge 11cfa76 into 3d5fb39
Browse files Browse the repository at this point in the history
  • Loading branch information
HotThoughts authored Feb 28, 2022
2 parents 3d5fb39 + 11cfa76 commit 04d84d6
Show file tree
Hide file tree
Showing 3 changed files with 12 additions and 1 deletion.
4 changes: 4 additions & 0 deletions .github/configs/tf-cuda.json
Original file line number Diff line number Diff line change
Expand Up @@ -12,6 +12,10 @@
{
"TF": "2.6",
"IMAGE_TAG": "cuda-11.2.0-cudnn8"
},
{
"TF": "2.7",
"IMAGE_TAG": "cuda-11.2.0-cudnn8"
}
]
}
7 changes: 7 additions & 0 deletions .github/workflows/ci-model-regression.yml
Original file line number Diff line number Diff line change
Expand Up @@ -160,6 +160,13 @@ jobs:
echo "GitHub runner image tag for TensorFlow ${{ env.TF_VERSION }} is ${GH_RUNNER_IMAGE_TAG}"
echo GH_RUNNER_IMAGE_TAG=$GH_RUNNER_IMAGE_TAG >> $GITHUB_ENV
- name: Send warning if CUDA image tag for the latest TF version has not configured yet
if: env.GH_RUNNER_IMAGE_TAG == 'latest'
env:
TF_CUDA_FILE: ./github/config/tf-cuda.json
run: |-
echo "::warning file=${TF_CUDA_FILE},line=3,col=1,endColumn=3::Missing cuda config for tf ${{ env.TF_VERSION }}. If you are not sure how to config CUDA, please reach out to infrastructure."
- name: Render deployment template
run: |-
export GH_RUNNER_IMAGE_TAG=${{ env.GH_RUNNER_IMAGE_TAG }}
Expand Down
2 changes: 1 addition & 1 deletion poetry.lock

Some generated files are not rendered by default. Learn more about how customized files appear on GitHub.

0 comments on commit 04d84d6

Please sign in to comment.