-
Notifications
You must be signed in to change notification settings - Fork 0
/
accelerator_naming.py
30 lines (30 loc) · 1.11 KB
/
accelerator_naming.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
NAME_MAP = {
"NVIDIA A100 80GB PCIe": "A100-80GB",
"NVIDIA A100-PCIE-40GB": "A100-40GB",
"NVIDIA A100-SXM4-80GB": "A100-80GB-SXM4",
"NVIDIA A30": "A30",
"NVIDIA A40": "A40",
"NVIDIA GeForce RTX 3070": "RTX3070",
"NVIDIA GeForce RTX 3080": "RTX3080",
"NVIDIA GeForce RTX 3080 Ti": "RTX3080Ti",
"NVIDIA GeForce RTX 3090": "RTX3090",
"NVIDIA GeForce RTX 3090 Ti": "RTX3090Ti",
"NVIDIA GeForce RTX 4070 Ti": "RTX4070Ti",
"NVIDIA GeForce RTX 4080": "RTX4080",
"NVIDIA GeForce RTX 4090": "RTX4090",
"NVIDIA H100 80GB HBM3": "H100-80GB-HBM3",
"NVIDIA H100 PCIe": "H100-PCIe",
"NVIDIA L4": "L4",
"NVIDIA L40": "L40",
"NVIDIA RTX 6000 Ada Generation": "RTX6000-Ada",
"NVIDIA RTX A4000": "RTXA4000",
"NVIDIA RTX A4500": "RTXA4500",
"NVIDIA RTX A5000": "RTXA5000",
"NVIDIA RTX A6000": "RTXA6000",
"Quadro RTX 5000": "RTX5000",
"Tesla V100-FHHL-16GB": "V100-16GB-FHHL",
"V100-SXM2-16GB": "V100-16GB-SXM2",
"NVIDIA RTX 4000 SFF Ada Generation": "RTX4000-Ada-SFF",
"NVIDIA RTX A2000": "RTXA2000",
"Tesla V100-PCIE-16GB": "V100-16GB-PCIe",
}