change nv_peer_mem_version

2024-07-14 09:11:41 +08:00
parent e68097db77
commit c6a09c405f
1 changed files with 1 additions and 1 deletions
--- 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"