chore: revert to version 3 tasks format for SSH support
This commit is contained in:
1 parent
f5dd3e5874
commit
c6ff7be55a
1 file changed
+14
-18
@@ -1,18 +1,14 @@
|
|||||||
$:
|
version: "3"
|
||||||
vscode:
|
tasks:
|
||||||
- docker:
|
- name: "训练小狸AI-Epoch100"
|
||||||
build: .ide/Dockerfile
|
image:
|
||||||
runner:
|
dockerfile: .ide/Dockerfile
|
||||||
cpus: 8
|
resources:
|
||||||
tags: cnb:arch:amd64:gpu
|
cpu: 8
|
||||||
services:
|
memory: 32G
|
||||||
- vscode
|
gpu: 1
|
||||||
- docker
|
env:
|
||||||
env:
|
CUDA_VISIBLE_DEVICES: "0"
|
||||||
CUDA_VISIBLE_DEVICES: "0"
|
PYTHONUNBUFFERED: "1"
|
||||||
PYTHONUNBUFFERED: "1"
|
commands:
|
||||||
stages:
|
- cd /workspace && nvidia-smi && echo "=== GPU就绪 ===" && 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
|
|
||||||
Reference in new issue
Block a user