small fixes

This commit is contained in:
Fangjun Kuang 2023-07-27 18:16:39 +08:00
parent e83b96059c
commit 4155f39d1e
2 changed files with 3 additions and 1 deletions

View File

@ -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:

View File

@ -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 \