feat: add CNB cloud training config (GPU Dockerfile + .cnb.yml + train_cnb.py)
This commit is contained in:
1 parent
eb7717e9ce
commit
456c443e9c
4 files changed
+223
-1
No files matched your search
@@ -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()}')"
|
||||
Reference in new issue
Block a user