From cf150580c6dfe02adbe58ba0d4dd6f1cae1f7854 Mon Sep 17 00:00:00 2001 From: vfdev-5 Date: Mon, 1 Apr 2024 12:49:49 +0200 Subject: [PATCH] Updated pytorch version in docker cfg to 2.2.2-cuda12.1-cudnn8 --- docker/docker.cfg | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/docker/docker.cfg b/docker/docker.cfg index 3000cbc1732..5c48d9f83e3 100644 --- a/docker/docker.cfg +++ b/docker/docker.cfg @@ -1,4 +1,4 @@ [DEFAULT] -build_docker_image_pytorch_version = 2.2.1-cuda12.1-cudnn8 +build_docker_image_pytorch_version = 2.2.2-cuda12.1-cudnn8 build_docker_image_hvd_version = v0.28.1 build_docker_image_msdp_version = v0.14.0