diff --git a/scrape.py b/scrape.py index c532cd0..a1706a7 100755 --- a/scrape.py +++ b/scrape.py @@ -534,6 +534,8 @@ def add_pretty_names(instances): "g3": "G3 Graphics GPU", "g4": "G4 Graphics and Machine Learning GPU", "g5": "G5 Graphics and Machine Learning GPU", + "g6": "G6 Graphics and Machine Learning GPU", + "gr6": "Gr6 Graphics and Machine Learning GPU High RAM ratio", "p2": "P2 General Purpose GPU", "p3": "P3 High Performance GPU", "p4d": "P4D Highest Performance GPU", @@ -817,6 +819,78 @@ def add_gpu_info(instances): "cuda_cores": 76928, "gpu_memory": 192, }, + "g6.xlarge": { + # GPU core count found from the whitepaper + # https://images.nvidia.com/aem-dam/Solutions/Data-Center/l4/nvidia-ada-gpu-architecture-whitepaper-v2.1.pdf + "gpu_model": "NVIDIA L4", + "compute_capability": 8.9, + "gpu_count": 1, + "cuda_cores": 7424, + "gpu_memory": 24, + }, + "g6.2xlarge": { + "gpu_model": "NVIDIA L4", + "compute_capability": 8.9, + "gpu_count": 1, + "cuda_cores": 7424, + "gpu_memory": 24, + }, + "g6.4xlarge": { + "gpu_model": "NVIDIA L4", + "compute_capability": 8.9, + "gpu_count": 1, + "cuda_cores": 7424, + "gpu_memory": 24, + }, + "g6.8xlarge": { + "gpu_model": "NVIDIA L4", + "compute_capability": 8.9, + "gpu_count": 1, + "cuda_cores": 7424, + "gpu_memory": 24, + }, + "gr6.4xlarge": { + "gpu_model": "NVIDIA L4", + "compute_capability": 8.9, + "gpu_count": 1, + "cuda_cores": 7424, + "gpu_memory": 24, + }, + "gr6.8xlarge": { + "gpu_model": "NVIDIA L4", + "compute_capability": 8.9, + "gpu_count": 1, + "cuda_cores": 7424, + "gpu_memory": 24, + }, + "g6.16xlarge": { + "gpu_model": "NVIDIA L4", + "compute_capability": 8.9, + "gpu_count": 1, + "cuda_cores": 7424, + "gpu_memory": 24, + }, + "g6.12xlarge": { + "gpu_model": "NVIDIA L4", + "compute_capability": 8.9, + "gpu_count": 4, + "cuda_cores": 29696, + "gpu_memory": 96, + }, + "g6.24xlarge": { + "gpu_model": "NVIDIA L4", + "compute_capability": 8.9, + "gpu_count": 4, + "cuda_cores": 29696, + "gpu_memory": 96, + }, + "g6.48xlarge": { + "gpu_model": "NVIDIA L4", + "compute_capability": 8.9, + "gpu_count": 8, + "cuda_cores": 59392, + "gpu_memory": 192, + }, "p4d.24xlarge": { "gpu_model": "NVIDIA A100", "compute_capability": 8.0,