update nv_peer_mem to 1.3

2024-07-14 08:45:22 +08:00
parent 158594cf3c
commit c32a973701
1 changed files with 1 additions and 0 deletions
--- 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"