chore: add vscode service to enable SSH in workspace

.cnb.yml

@@ -1,14 +1,18 @@
-version: "3"
+$:
-tasks:
+  vscode:
-  - name: "训练小狸AI-Epoch100"
+    - docker:
-    image:
+        build: .ide/Dockerfile
-      dockerfile: .ide/Dockerfile
+      runner:
-    resources:
+        cpus: 8
-      cpu: 8
+        tags: cnb:arch:amd64:gpu
-      memory: 32G
+      services:
-      gpu: 1
+        - vscode
+        - docker
       env:
         CUDA_VISIBLE_DEVICES: "0"
         PYTHONUNBUFFERED: "1"
-    commands:
+      stages:
-      - cd /workspace && nvidia-smi && python3 scripts/train_cnb.py --epochs 100 --batch_size 16
+        - name: ls
+          script: ls -al
+        - name: 开始训练
+          script: cd /workspace && nvidia-smi && python3 scripts/train_cnb.py --epochs 100 --batch_size 16