feat: add CNB cloud training config (GPU Dockerfile + .cnb.yml + train_cnb.py)

This commit is contained in:
shuiyu1123 committed 2026-06-18 18:09:59 +00:00
1 parent eb7717e9ce
commit 456c443e9c
4 files changed
+223 -1

No files matched your search

+13
View File
@@ -0,0 +1,13 @@
FROM pytorch/pytorch:2.1.0-cuda12.1-cudnn8-runtime
# 基础工具
RUN apt-get update && apt-get install -y git curl wget nano && rm -rf /var/lib/apt/lists/*
# tqdm用于进度条
RUN pip install --no-cache-dir tqdm numpy requests
# 设置工作目录
WORKDIR /workspace
# 验证CUDA
RUN python3 -c "import torch; print(f'PyTorch {torch.__version__}, CUDA: {torch.cuda.is_available()}')"