We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent 32701b5 commit a5b08eaCopy full SHA for a5b08ea
.github/workflows/docker_builder.yml
@@ -39,6 +39,7 @@ jobs:
39
env:
40
DOCKER_TAG: ${{ env.DOCKER_REGISTRY }}/${{ env.CONTAINER_NAME }}
41
run: |
42
+ python3 -m pip install pyyaml
43
TRT_VERSION=$(python3 -c "import versions; versions.tensorrt_version()")
44
echo "TRT VERSION = ${TRT_VERSION}"
45
CUDNN_VERSION=$(python3 -c "import versions; versions.cudnn_version()")
0 commit comments