From c6a09c405f69d9d706fec019b3c3bdbea5e553fd Mon Sep 17 00:00:00 2001 From: lingyuzeng Date: Sun, 14 Jul 2024 09:11:41 +0800 Subject: [PATCH] change nv_peer_mem_version --- finetune/docker-compose_update.yml | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/finetune/docker-compose_update.yml b/finetune/docker-compose_update.yml index 076d70e..ba6d842 100644 --- a/finetune/docker-compose_update.yml +++ b/finetune/docker-compose_update.yml @@ -20,7 +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" + NV_PEER_MEM_VERSION: "1.2" CUDA_VERSION: "12.1.0" PYTORCH_VERSION: "2.3.0" TORCHVISION_VERSION: "0.18.0"