From 9e6839f6f3f8307722768f2444ab8a294908167e Mon Sep 17 00:00:00 2001
From: Ronghang Hu <ronghang.hu@gmail.com>
Date: Wed, 11 Dec 2024 17:23:19 +0000
Subject: [PATCH] update demo docker version to PyTorch 2.5.1

---
 backend.Dockerfile | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/backend.Dockerfile b/backend.Dockerfile
index adec61d..54a3296 100644
--- a/backend.Dockerfile
+++ b/backend.Dockerfile
@@ -1,4 +1,4 @@
-ARG BASE_IMAGE=pytorch/pytorch:2.3.1-cuda12.1-cudnn8-runtime
+ARG BASE_IMAGE=pytorch/pytorch:2.5.1-cuda12.1-cudnn9-runtime
 ARG MODEL_SIZE=base_plus
 
 FROM ${BASE_IMAGE}