update

2024-07-02 06:08:08 +00:00
parent 4081db7ce0
commit 54a2227837
4 changed files with 67 additions and 5 deletions
--- a/finetune/docker-compose_nccl.yml
+++ b/finetune/docker-compose_nccl.yml
@@ -8,6 +8,8 @@ services:
    environment:
      - PORT=1998
      - PASS=P@88w0rd
+      - NVIDIA_VISIBLE_DEVICES=all
+      - NVIDIA_DRIVER_CAPABILITIES=compute,utility
    volumes:
      - ./id_rsa_finetune:/root/.ssh/id_rsa
      - ./id_rsa.pub:/root/.ssh/id_rsa.pub