From 4155f39d1e863ac0577c11957a423671fca462cc Mon Sep 17 00:00:00 2001 From: Fangjun Kuang Date: Thu, 27 Jul 2023 18:16:39 +0800 Subject: [PATCH] small fixes --- .github/workflows/build-docker-image.yml | 2 ++ docker/torch1.13.0-cuda11.6-cudnn8.dockerfile | 2 +- 2 files changed, 3 insertions(+), 1 deletion(-) diff --git a/.github/workflows/build-docker-image.yml b/.github/workflows/build-docker-image.yml index 2e1dbf8aa..04cf8af13 100644 --- a/.github/workflows/build-docker-image.yml +++ b/.github/workflows/build-docker-image.yml @@ -1,3 +1,5 @@ +# see also +# https://docs.github.com/en/actions/publishing-packages/publishing-docker-images#publishing-images-to-github-packages name: Build docker image on: push: diff --git a/docker/torch1.13.0-cuda11.6-cudnn8.dockerfile b/docker/torch1.13.0-cuda11.6-cudnn8.dockerfile index 3f8fd0d37..c85d3cf17 100644 --- a/docker/torch1.13.0-cuda11.6-cudnn8.dockerfile +++ b/docker/torch1.13.0-cuda11.6-cudnn8.dockerfile @@ -1,4 +1,4 @@ -FROM pytorch/pytorch:1.13.0-cuda11.6-cudnn8-devel +FROM pytorch/pytorch:1.13.0-cuda11.6-cudnn8-runtime # Install dependencies RUN pip install --no-cache-dir \