From c32a973701eb2ef26c78042c9f7b7426cb85c897 Mon Sep 17 00:00:00 2001 From: lingyuzeng Date: Sun, 14 Jul 2024 08:45:22 +0800 Subject: [PATCH] update nv_peer_mem to 1.3 --- finetune/docker-compose_update.yml | 1 + 1 file changed, 1 insertion(+) diff --git a/finetune/docker-compose_update.yml b/finetune/docker-compose_update.yml index c8247ba..076d70e 100644 --- a/finetune/docker-compose_update.yml +++ b/finetune/docker-compose_update.yml @@ -20,6 +20,7 @@ services: dockerfile: Dockerfile.update args: # PyTorch版本、Python版本与pytorch_lightning版本的对应关系表 https://blog.csdn.net/qq_41813454/article/details/137421822 PYTHON_VERSION: "3.10" + NV_PEER_MEM_VERSION: "1.3" CUDA_VERSION: "12.1.0" PYTORCH_VERSION: "2.3.0" TORCHVISION_VERSION: "0.18.0"