chore: Add pyyaml import to GHA Docker job (#2170)

pytorch · Aug 3, 2023 · 0527edd · 0527edd
1 parent 32701b5
commit 0527edd
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/.github/workflows/docker_builder.yml b/.github/workflows/docker_builder.yml
@@ -39,6 +39,7 @@ jobs:
       env:
         DOCKER_TAG: ${{ env.DOCKER_REGISTRY }}/${{ env.CONTAINER_NAME }}
       run: |
+        python3 -m pip install pyyaml
         TRT_VERSION=$(python3 -c "import versions; versions.tensorrt_version()")
         echo "TRT VERSION = ${TRT_VERSION}"
         CUDNN_VERSION=$(python3 -c "import versions; versions.cudnn_version()")