Skip to content

Commit c9da5b9

Browse files
authored
Add missing GeForce RTX {2..5}xxx GPUs to KNOWN_NVIDIA_GPUS (#229)
Note, 2xxx "super" suffix is stylized as "Super", while 4xxx as "SUPER", but we use ALLCAPS for consistency in both cases
1 parent 99254da commit c9da5b9

1 file changed

Lines changed: 27 additions & 0 deletions

File tree

src/gpuhunt/_internal/constraints.py

Lines changed: 27 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -182,14 +182,41 @@ def is_nvidia_superchip(gpu_name: str) -> bool:
182182
NvidiaGPUInfo(name="L40", memory=48, compute_capability=(8, 9)),
183183
NvidiaGPUInfo(name="L40S", memory=48, compute_capability=(8, 9)),
184184
NvidiaGPUInfo(name="P100", memory=16, compute_capability=(6, 0)),
185+
NvidiaGPUInfo(name="RTX2060", memory=6, compute_capability=(7, 5)),
186+
NvidiaGPUInfo(name="RTX2060", memory=12, compute_capability=(7, 5)),
187+
NvidiaGPUInfo(name="RTX2060SUPER", memory=8, compute_capability=(7, 5)),
188+
NvidiaGPUInfo(name="RTX2070", memory=8, compute_capability=(7, 5)),
189+
NvidiaGPUInfo(name="RTX2070SUPER", memory=8, compute_capability=(7, 5)),
190+
NvidiaGPUInfo(name="RTX2080", memory=8, compute_capability=(7, 5)),
191+
NvidiaGPUInfo(name="RTX2080SUPER", memory=8, compute_capability=(7, 5)),
192+
NvidiaGPUInfo(name="RTX2080Ti", memory=11, compute_capability=(7, 5)),
193+
NvidiaGPUInfo(name="RTX3050", memory=6, compute_capability=(8, 6)),
194+
NvidiaGPUInfo(name="RTX3050", memory=8, compute_capability=(8, 6)),
185195
NvidiaGPUInfo(name="RTX3060", memory=8, compute_capability=(8, 6)),
186196
NvidiaGPUInfo(name="RTX3060", memory=12, compute_capability=(8, 6)),
187197
NvidiaGPUInfo(name="RTX3060Ti", memory=8, compute_capability=(8, 6)),
198+
NvidiaGPUInfo(name="RTX3070", memory=8, compute_capability=(8, 6)),
188199
NvidiaGPUInfo(name="RTX3070Ti", memory=8, compute_capability=(8, 6)),
189200
NvidiaGPUInfo(name="RTX3080", memory=10, compute_capability=(8, 6)),
190201
NvidiaGPUInfo(name="RTX3080Ti", memory=12, compute_capability=(8, 6)),
191202
NvidiaGPUInfo(name="RTX3090", memory=24, compute_capability=(8, 6)),
203+
NvidiaGPUInfo(name="RTX3090Ti", memory=24, compute_capability=(8, 6)),
204+
NvidiaGPUInfo(name="RTX4060", memory=8, compute_capability=(8, 9)),
205+
NvidiaGPUInfo(name="RTX4060Ti", memory=8, compute_capability=(8, 9)),
206+
NvidiaGPUInfo(name="RTX4060Ti", memory=16, compute_capability=(8, 9)),
207+
NvidiaGPUInfo(name="RTX4070", memory=12, compute_capability=(8, 9)),
208+
NvidiaGPUInfo(name="RTX4070SUPER", memory=12, compute_capability=(8, 9)),
209+
NvidiaGPUInfo(name="RTX4070Ti", memory=12, compute_capability=(8, 9)),
210+
NvidiaGPUInfo(name="RTX4070TiSUPER", memory=16, compute_capability=(8, 9)),
211+
NvidiaGPUInfo(name="RTX4080", memory=16, compute_capability=(8, 9)),
212+
NvidiaGPUInfo(name="RTX4080SUPER", memory=16, compute_capability=(8, 9)),
192213
NvidiaGPUInfo(name="RTX4090", memory=24, compute_capability=(8, 9)),
214+
NvidiaGPUInfo(name="RTX5060", memory=8, compute_capability=(12, 0)),
215+
NvidiaGPUInfo(name="RTX5060Ti", memory=8, compute_capability=(12, 0)),
216+
NvidiaGPUInfo(name="RTX5060Ti", memory=16, compute_capability=(12, 0)),
217+
NvidiaGPUInfo(name="RTX5070", memory=12, compute_capability=(12, 0)),
218+
NvidiaGPUInfo(name="RTX5070Ti", memory=16, compute_capability=(12, 0)),
219+
NvidiaGPUInfo(name="RTX5080", memory=16, compute_capability=(12, 0)),
193220
NvidiaGPUInfo(name="RTX5090", memory=32, compute_capability=(12, 0)),
194221
NvidiaGPUInfo(name="RTX6000", memory=24, compute_capability=(7, 5)),
195222
NvidiaGPUInfo(name="RTX2000Ada", memory=16, compute_capability=(8, 9)),

0 commit comments

Comments
 (0)