Tensorflow:它可以在GTX 1070(8GB)上运行ResNet-101吗?

时间:2016-09-06 07:22:21

标签: tensorflow gpu nvidia

使用tensorflow,是否足以运行ResNet-101的GTX 1070(8GB RAM)?

根据ResNet论文,ResNet比VGG-16“更简单”。由于VGG-16在目前的4 GB GPU(GTX 960)中已经运行良好,因此更大的8GB 1070 GPU应该对ResNet没有任何问题吗?

1 个答案:

答案 0 :(得分:3)

根据this benchmark

  

我们对所有型号进行基准测试,其中小批量尺寸为16,图像尺寸为224 x 224;这允许在模型之间进行直接比较,并允许除了ResNet-200模型以外的所有模型在GTX 1080上运行,而GTX 1080只有8GB内存。

所有ResNet型号,但200层深,适合8GB 1080 GTX。由于1070 GTX还有8GB的内存,ResNets最多可以容纳152层,而不是更深层的内存。