[CI] Update docker image for Publish workflow (#483)

Update docker image for Publish workflow to corresponds Tests workflow

Author: vadiklyutiy

.github/workflows/publish-pypi.yaml

@@ -58,7 +58,7 @@ jobs:
       matrix: ${{ fromJSON(needs.list-test-dirs.outputs.matrix) }}
     runs-on: arc-runner-set
     container:
-      image: nvidia/cuda:11.8.0-devel-ubuntu20.04
+      image: nvidia/cuda:12.6.2-devel-ubuntu22.04
     steps:
       - name: Install dependencies via apt
         run: |
@@ -93,7 +93,7 @@ jobs:
   build-docs:
     runs-on: arc-runner-set
     container:
-      image: nvidia/cuda:11.8.0-devel-ubuntu20.04
+      image: nvidia/cuda:12.6.2-devel-ubuntu22.04
     steps:
       - name: Install dependencies via apt
         run: |