arn-v2 / environment_v2.json
AnishPathak2202's picture
ARN v2: model card, checkpoints, environment
356c945 verified
Raw History Blame Contribute Delete
455 Bytes
{
"python": "3.10.11",
"platform": "Windows-10-10.0.26300-SP0",
"torch": "2.12.0.dev20260408+cu128",
"torch_cuda": "12.8",
"cudnn": 92000,
"transformers": "5.5.3",
"numpy": "1.26.4",
"gpu": "NVIDIA GeForce RTX 5050 Laptop GPU",
"gpu_capability": "sm_120",
"gpu_memory_gb": 8.52,
"cpu": "Intel64 Family 6 Model 191 Stepping 2, GenuineIntel",
"note": "torch is a nightly/dev build; pin this exact version to reproduce"
}