update

megadna/docker-compose_pytorch2.3.yml

@@ -1,11 +1,12 @@
 version: '3.8'
 
 services:
-  ubuntu-finetune:
+  ubuntu-megadna:
     build: 
       context: .
       dockerfile: Dockerfile
       args: # PyTorch版本、Python版本与pytorch_lightning版本的对应关系表 https://blog.csdn.net/qq_41813454/article/details/137421822
+        CONDA_ENV_NAME: megadna
         CUDA_VERSION: 12.1.0
         PYTORCH_VERSION: 2.3.0
         TORCHVISION_VERSION: 0.18.0
@@ -19,19 +20,19 @@ services:
         SETUPTOOLS_VERSION: "69.5.1"
     volumes:
       - ./src:/bbtft
-    container_name: ubuntu-finetune
+    container_name: ubuntu-megadna
     pull_policy: if_not_present
     tty: true
     restart: unless-stopped
     image: hotwa/deepspeed:test
     shm_size: '32gb'
     ports:
-      - 3227:2222
+      - 3237:22
     environment:
       - NVIDIA_VISIBLE_DEVICES=all
       - NVIDIA_DRIVER_CAPABILITIES=compute,utility
     networks:
-      - network_finetune
+      - network_megadna
     deploy:
       resources:
         reservations:
@@ -41,5 +42,5 @@ services:
               capabilities: [gpu]
 
 networks:
-  network_finetune:
-    name: network_finetune
+  network_megadna:
+    name: network_megadna