From e1e9f63105fbf1c2d1dcf7e2d6059c3b5597858e Mon Sep 17 00:00:00 2001 From: cyntachs Date: Mon, 22 Apr 2024 10:14:23 -0400 Subject: [PATCH] updated pytorch to 2.2 updated to pytorch 2.2.2 with cuda 12.1 --- Dockerfile | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/Dockerfile b/Dockerfile index 4bf12b7..f56ac0f 100644 --- a/Dockerfile +++ b/Dockerfile @@ -1,6 +1,6 @@ -FROM pytorch/pytorch:2.1.0-cuda12.1-cudnn8-runtime +FROM pytorch/pytorch:2.2.2-cuda12.1-cudnn8-runtime -LABEL version="1.2-cuda12.1" maintainer="Cyntachs" +LABEL version="py2.2-cuda12.1" maintainer="Cyntachs" ARG DEBIAN_FRONTEND=noninteractive ENV TZ="America/New_York"