fix: add vscode-server dir to Dockerfile, use vscode event for SSH
This commit is contained in:
1 parent
c6ff7be55a
commit
7b47faac50
2 files changed
+28
-17
No files matched your search
@@ -1,14 +1,21 @@
|
||||
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
|
||||
$:
|
||||
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: 开始训练
|
||||
script: |
|
||||
cd /workspace
|
||||
nvidia-smi
|
||||
echo "=== GPU就绪,开始训练 ==="
|
||||
python3 scripts/train_cnb.py --epochs 100 --batch_size 16
|
||||
echo "=== 训练完成 ==="
|
||||
Reference in new issue
Block a user