Updated pytorch version in docker cfg to 2.2.2-cuda12.1-cudnn8 (#3227)

pytorch · Apr 1, 2024 · 5bf73e2 · 5bf73e2
1 parent b828059
commit 5bf73e2
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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