图像增强需要多少GPU？

Question

我在找一个GPU来训练我的模特。我所跟踪的大多数论文使用了2张或更多的gtx 1050 gtx卡或更高的卡。(MIRNet，EnlightenGAN)我需要多少GPU的力量来训练我的模型。在我开始我的研究之前，我需要估计我的资源。我的预算很紧，不超过1200美元。联想军团5( ryzen 7，16 6b内存，1660ti 6b)就足够了吗？我应该选择RTX 2060或更高的吗？或者你能提出其他选择吗。

我想知道足够训练那些模特了吗？谢谢!

Answer 1

我会和RTX一起去的。如果我要设计深度学习模型，我更喜欢GPU中更多的核心，RTX是赢家。

不要忘记RTX中存在的张量核心，这使得它在深度学习过程中脱颖而出。(更快的张量操作)

如果你能负担得起的话，就和RTX一起去吧。

 ╔════════════════╦════════════╦══════════╗
 ║      GPU       ║ GTX-1660Ti ║ RTX-2060 ║
 ╠════════════════╬════════════╬══════════╣
 ║ GPU-cores      ║ 1536       ║ 1920     ║
 ║ Tensor-cores   ║ N/A        ║ 240      ║
 ║ RT-cores       ║ N/A        ║ 30       ║
 ║ Texture-units  ║ 120        ║ 120      ║
 ║ Render-outputs ║ 48         ║ 48       ║
 ║ Base-clock     ║ 1500       ║ 1365     ║
 ║ Boost-clock    ║ 1770       ║ 1680     ║
 ║ Mem-bus-width  ║ 192-bit    ║ 192-bit  ║
 ║ Mem-bandwidth  ║ 288GB/s    ║ 336GB/s  ║
 ║ Mem-capacity   ║ 6GB        ║ 6GB      ║
 ║ TDP            ║ 120W       ║ 160W     ║
 ╚════════════════╩════════════╩══════════╝