diff --git a/.cnb.yml b/.cnb.yml
index e51a4b4..9d5a9c4 100644
--- a/.cnb.yml
+++ b/.cnb.yml
@@ -1,18 +1,14 @@
-$:
-  vscode:
-    - docker:
-        build: .ide/Dockerfile
-      runner:
-        cpus: 8
-        tags: cnb:arch:amd64:gpu
-      services:
-        - vscode
-        - docker
-      env:
-        CUDA_VISIBLE_DEVICES: "0"
-        PYTHONUNBUFFERED: "1"
-      stages:
-        - name: ls
-          script: ls -al
-        - name: 开始训练
-          script: cd /workspace && nvidia-smi && python3 scripts/train_cnb.py --epochs 100 --batch_size 16
\ No newline at end of file
+version: "3"
+tasks:
+  - name: "训练小狸AI-Epoch100"
+    image:
+      dockerfile: .ide/Dockerfile
+    resources:
+      cpu: 8
+      memory: 32G
+      gpu: 1
+    env:
+      CUDA_VISIBLE_DEVICES: "0"
+      PYTHONUNBUFFERED: "1"
+    commands:
+      - cd /workspace && nvidia-smi && echo "=== GPU就绪 ===" && python3 scripts/train_cnb.py --epochs 100 --batch_size 16
\ No newline at end of file