chore: add vscode service to enable SSH in workspace

This commit is contained in:
shuiyu1123 committed 2026-06-18 19:53:09 +00:00
1 parent ac07c7f06a
commit f5dd3e5874
1 file changed
+18 -14
+18 -14
View File
@@ -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
env: - docker
CUDA_VISIBLE_DEVICES: "0" env:
PYTHONUNBUFFERED: "1" CUDA_VISIBLE_DEVICES: "0"
commands: PYTHONUNBUFFERED: "1"
- cd /workspace && nvidia-smi && python3 scripts/train_cnb.py --epochs 100 --batch_size 16 stages:
- name: ls
script: ls -al
- name: 开始训练
script: cd /workspace && nvidia-smi && python3 scripts/train_cnb.py --epochs 100 --batch_size 16