Titan X Pascal在VGG16上跑得比基准慢得多
我有一块Titan X Pascal,Intel i5-6600,16GB RAM,在Ubuntu 14.04上运行torch7。 Nvidia驱动版本为375.20,CUDA Toolkit为8.0,cuDNN为v5.1。
我使用与 此基准测试 相同的VGG16网络进行了相同的测试(通过loadcaffe导入)。但是,对于前向传递,我的设置需要80ms,这是基准测试需要的时间的两倍。
我还生成了一批16个RGB通道,尺寸为224x224的图像。相关代码如下:
local model = loadcaffe.load("/home/.../Models/VGG16/VGG_ILSVRC_16_layers_deploy.prototxt",
"/home/.../Models/VGG16/VGG_ILSVRC_16_layers.caffemodel",
"cudnn")
for i=1, 50 do
local input = torch.randn(16, 3, 224, 224):type("torch.CudaTensor")
cutorch.synchronize()
local timer = torch.Timer()
model:forward(input)
cutorch.synchronize()
local deltaT = timer:time().real
print("前向传递时间:" .. deltaT)
end
输出如下:
前向传递时间:0.96536016464233 前向传递时间:0.10063600540161 前向传递时间:0.096444129943848 前向传递时间:0.089151859283447 前向传递时间:0.082037925720215 前向传递时间:0.082045078277588 前向传递时间:0.079913139343262 前向传递时间:0.080273866653442 前向传递时间:0.080694913864136 前向传递时间:0.082727193832397 前向传递时间:0.082070827484131 前向传递时间:0.079407930374146 前向传递时间:0.080456018447876 前向传递时间:0.083559989929199 前向传递时间:0.082060098648071 前向传递时间:0.081624984741211 前向传递时间:0.080413103103638 前向传递时间:0.083755016326904 前向传递时间:0.083209037780762 ...
我需要做些什么额外的工作才能获得那种速度?还是我做错了什么?还是因为我使用的是Ubuntu 14.04,而不是Ubuntu 16.04(尽管在基准测试中,运行在Ubuntu 14.04上的GTX 1080也只需要60ms)?
- Lua 虚拟机加密load(string.dump(function)) 后执行失败问题如何解决
- 我想创建一个 Nginx 规则,禁止访问
- 如何将两个不同的lua文件合成一个 东西有点长 大佬请耐心看完 我是小白研究几天了都没搞定
- 如何在roblox studio中1:1导入真实世界的地形?
- 求解,lua_resume的第二次调用继续执行协程问题。
- 【上海普陀区】内向猫网络招募【Skynet游戏框架Lua后端程序员】
- SF爱好求教:如何用lua实现游戏内调用数据库函数实现账号密码注册?
- Lua实现网站后台开发
- LUA错误显式返回,社区常见的规约是怎么样的
- lua5.3下载库失败
- 请问如何实现文本框内容和某个网页搜索框内容连接,并把网页输出来的结果反馈到另外一个文本框上
- lua lanes多线程使用
- 一个kv数据库
- openresty 有没有比较轻量的 docker 镜像
- 想问一下,有大佬用过luacurl吗
- 在Lua执行过程中使用Load函数出现问题
- 为什么 neovim 里没有显示一些特殊字符?
- Lua比较两个表的值(不考虑键的顺序)
- 有个lua简单的项目,外包,有意者加微信 liuheng600456详谈,最好在成都
- 如何在 Visual Studio 2022 中运行 Lua 代码?

我终于找到了解决方法。
我需要启用
cudnn.benchmark标志:cudnn.benchmark = true默认情况下它被设置为false,所以
cudnn不会选择最快的算法。现在我的前向时间大约是39ms。