generated from mathias/template-go-web
mathiasandClaude Opus 4.8
df910e4336
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>
22 lines
829 B
Python
22 lines
829 B
Python
"""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")
|