最近一直在搞神经网络,因为习惯了Windows的操作,一直在Windows下看文章写东西,也就顺势在Windows下做了代码编写和训练。一开始觉得速度也就那样,没太注意,但一次因为电脑比较忙拿同样的代码到别的装了Ubuntu系统的机子上面跑,才发现它们的效率意外的相去甚远。
猜想操作系统本身可能会影响网络训练效率,因此在同一台机子上测试了同样的代码在两个操作系统平台下的运行速度,希望看到的人能得到些许有用信息。
MXNet架构
实现内容为FCN全卷积神经网络,基本照抄的沐神的公开课的代码。
Windows平台:
Ubuntu平台
Pytorch架构
实现内容同样是FCN全卷积神经网络,是本人自己写的代码。
Windows平台:
Ubuntu平台
因为近期只做了这两个程序,就对比了这两个,后面如果有新的会放上来。但从这两个程序的结果可以明显看出,Ubuntu平台或多或少还是要快一些,所以如果做大型训练,还是应该选择Ubuntu或者其他Linux平台更靠谱。