generated from mathias/template-go-web
train.py imports torch but the venv step only installed requirements.txt, which deliberately excludes torch (must come from the cu130 wheel index for koala's Blackwell sm_120, per the requirements.txt header). Baseline run died with ModuleNotFoundError: No module named 'torch'. Add the documented install.