chore: revert to version 3 tasks format for SSH support

This commit is contained in:
shuiyu1123 committed 2026-06-18 20:05:41 +00:00
1 parent f5dd3e5874
commit c6ff7be55a
1 file changed
+14 -18
+14 -18
View File
@@ -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
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