From 4ccd08b19a85ffd263cfb4cb0c1e9dbdfffcda1a Mon Sep 17 00:00:00 2001 From: jimmy Date: Mon, 4 Nov 2024 21:07:27 +0800 Subject: [PATCH] add github action build docker image --- docker/Dockerfile | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/docker/Dockerfile b/docker/Dockerfile index 972da45..1fe523a 100644 --- a/docker/Dockerfile +++ b/docker/Dockerfile @@ -1,4 +1,4 @@ -FROM pytorch/pytorch:2.4.1-cuda12.1-cudnn9-runtime +FROM pytorch/pytorch:2.5.1-cuda12.1-cudnn9-runtime # Ignore all interactive dialog during apt-get update ENV DEBIAN_FRONTEND=noninteractive