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"