完整源码可复制、可下载。先理解执行位置,再提交。
Slurm 脚本从项目根目录提交,并先创建 logs/。训练示例需要 Python 环境;不要直接在登录节点运行训练。
logs/
标准库小测试,无需先安装 PyTorch。
slurm/debug_cpu.sbatch
显式调用虚拟环境,记录输出,错误时退出。
slurm/train_cpu.sbatch
3 个学习率 × 5 个种子,默认最多并发 3 项。
slurm/train_array.sbatch
先检查 CUDA,再执行示例;需要 GPU 环境。
slurm/train_gpu.sbatch
输出所在节点和 Slurm 参数,保存一个 JSON。
examples/hello_hpc.py
小型回归训练,保存指标、元信息与 checkpoint。
examples/train_demo.py
检查 CUDA 可见性并运行小型矩阵计算。
examples/gpu_check.py
合并到本机 SSH 配置,替换用户名并核验密钥。
templates/ssh_config.example
CPU 示例依赖;安装方式参照环境章节。
examples/requirements.txt
输入关键词,搜索教程和完整脚本。