From f17fc79ac498e54cf9ef2994bfb4e276beab2217 Mon Sep 17 00:00:00 2001 From: vfdev Date: Fri, 14 Jun 2024 15:07:08 +0200 Subject: [PATCH] Updated docker pth version --- docker/docker.cfg | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/docker/docker.cfg b/docker/docker.cfg index 5c48d9f83e3..dd43c35e7df 100644 --- a/docker/docker.cfg +++ b/docker/docker.cfg @@ -1,4 +1,4 @@ [DEFAULT] -build_docker_image_pytorch_version = 2.2.2-cuda12.1-cudnn8 +build_docker_image_pytorch_version = 2.3.1-cuda12.1-cudnn8 build_docker_image_hvd_version = v0.28.1 build_docker_image_msdp_version = v0.14.0