使用CUDA 7.5培训GeForce GTX Titan X的时间

Question

我使用CUDA 7.5（Ubuntu 14）在GeForce GTX Titan X上运行Caffe库。我不确定是否为我的设置正确配置了Caffe。我的数据集由256 x 256像素（3个通道），100000个训练/ 10000个测试样本组成。对于第一次测试，我使用AlexNet，new_height = 256，new_width = 256，crop_size = 227。使用batch_size = 256在一个Titan X上运行1000次训练迭代需要大约17分钟...这个硬件是不是太慢了？

非常感谢任何帮助和建议！

Answer 1

对一批256张图片运行1000次迭代： （256高* 256宽* 256批量* 1000次迭代* 3通道）字节/（（1024 * 1024）MB *（17 * 60）秒）= 47MBps计算速度。

以下内容可能会改善效果：

• 如果原始图像的分辨率较高，请尝试将它们预处理为256x256，从而减少硬盘中的大量像素读取。
• 使用Cudnn标志编译Caffe。这可能会使速度提高30％
• 尝试创建输入集的LMDB数据集，并使用LMDB数据进行培训。
• 尝试使用SSD而不是SATA硬盘。

Answer 2

不，不是。查看this链接了解Caffe性能和硬件配置。