Learning Center

import torch
import os

def check_gaashub():
    print("🚀 GaasHub Sanity Test Starting...")
    
    # 1. Check if connected via SCUDA
    is_connected = os.path.exists(os.path.expanduser("~/.scuda/.connected"))
    print(f"📍 Connection Status: {'✅ ACTIVE' if is_connected else '❌ NOT CONNECTED'}")
    
    # 2. Check GPU Visibility
    if torch.cuda.is_available():
        gpu_name = torch.cuda.get_device_name(0)
        vram = torch.cuda.get_device_properties(0).total_memory / 1e9
        print(f"🎬 GPU Detected: {gpu_name}")
        print(f"💾 Available VRAM: {vram:.2f} GB")
        
        # 3. Simple Matrix Multiplication Test
        print("⚡ Running compute test...")
        a = torch.randn(1000, 1000).cuda()
        b = torch.randn(1000, 1000).cuda()
        c = torch.matmul(a, b)
        print("✨ Computation Successful!")
    else:
        print("❌ No GPU detected. Make sure you are connected in the GaasHub app.")

if __name__ == "__main__":
    check_gaashub()