使用tensorflow,是否足以运行ResNet-101的GTX 1070(8GB RAM)?
根据ResNet论文,ResNet比VGG-16“更简单”。由于VGG-16在目前的4 GB GPU(GTX 960)中已经运行良好,因此更大的8GB 1070 GPU应该对ResNet没有任何问题吗?
答案 0 :(得分:3)
我们对所有型号进行基准测试,其中小批量尺寸为16,图像尺寸为224 x 224;这允许在模型之间进行直接比较,并允许除了ResNet-200模型以外的所有模型在GTX 1080上运行,而GTX 1080只有8GB内存。
所有ResNet型号,但200层深,适合8GB 1080 GTX。由于1070 GTX还有8GB的内存,ResNets最多可以容纳152层,而不是更深层的内存。