up

neoguojing · Aug 17, 2024 · 6bbec6e · 6bbec6e
1 parent d41df62
commit 6bbec6e
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/deploy/Dockerfile.base b/deploy/Dockerfile.base
@@ -1,5 +1,5 @@
 # 使用指定的基础镜像
-FROM pytorch/pytorch:2.2.2-cuda12.1-cudnn8-runtime
+FROM pytorch/pytorch:2.3.1-cuda12.1-cudnn8-runtime
 
 # 更新 apt-get 并安装依赖
 # RUN apt-get update && \