generated from mathias/template-go-web
chore(phase0): reproducible compute gate — torch cu130 + GPU smoke test
scripts/check_gpu.py verifies PyTorch cu130 sees the koala Blackwell GPU (sm_120) and computes — the Phase-0 prerequisite before any autoresearch experiment. Verified green: torch 2.12.1+cu130, RTX 5070, GPU matmul OK. Note: koala GPU is shared with the llama-swap LLM stack — run the autoresearch agent on iguana/berget so the card stays free for train.py. Co-Authored-By: Claude Opus 4.8 (1M context) <noreply@anthropic.com>
This commit is contained in:
@@ -0,0 +1,21 @@
|
||||
"""Phase-0 compute gate (brain wiki/jepa-fx/facts/autoresearch-integration-phase1):
|
||||
PyTorch cu130 must see the koala Blackwell GPU and compute before any experiment.
|
||||
|
||||
python scripts/check_gpu.py # exits 0 if the GPU is usable, 1 otherwise
|
||||
|
||||
Note: koala shares this 12GB card with the llama-swap LLM stack. The autoresearch
|
||||
agent should run on iguana/berget models so koala's GPU stays free for train.py.
|
||||
"""
|
||||
import sys
|
||||
import torch
|
||||
|
||||
print("torch", torch.__version__)
|
||||
if not torch.cuda.is_available():
|
||||
print("CUDA NOT AVAILABLE — gate BLOCKED")
|
||||
sys.exit(1)
|
||||
print("device:", torch.cuda.get_device_name(0))
|
||||
print("capability: sm_%d%d" % torch.cuda.get_device_capability(0))
|
||||
x = torch.randn(2000, 2000, device="cuda")
|
||||
(x @ x).sum().item()
|
||||
torch.cuda.synchronize()
|
||||
print("GPU matmul OK — Phase-0 compute gate GREEN")
|
||||
Reference in New Issue
Block a user