Set Compute Level for CI build (#10276)

This commit is contained in:
Nate Meyer 2024-03-07 06:59:33 -05:00 committed by GitHub
parent ccb5e05e3e
commit 8bfc0c98dc
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194

View File

@ -38,6 +38,8 @@ jobs:
tags: ${{ steps.setup.outputs.image-name }}-amd64 tags: ${{ steps.setup.outputs.image-name }}-amd64
cache-from: type=registry,ref=${{ steps.setup.outputs.cache-name }}-amd64 cache-from: type=registry,ref=${{ steps.setup.outputs.cache-name }}-amd64
- name: Build and push TensorRT (x86 GPU) - name: Build and push TensorRT (x86 GPU)
env:
COMPUTE_LEVEL: "50 60 70 80 90"
uses: docker/bake-action@v4 uses: docker/bake-action@v4
with: with:
push: true push: true