使用CUDA的Torch7(CUnn和CUtorch)- MNIST的Torch演示
2016-5-3 21:36:26
收藏:0
阅读:87
评论:0
我从github上下载了torch的演示程序(https://github.com/torch/demos/tree/master/train-a-digit-classifier)。这段代码在CPU模式下运行良好,但是我想要在CUDA 6.5的GPU模式下使用它。我已经安装了所有的先决条件和CUnn以及CUtorch,并根据我找到的文档和教程,添加了一些代码行使它工作:
开始时:
require 'cunn'
require 'cutorch'
并在模型声明后结束:
model:cuda()
但是,当我运行代码时,出现以下错误:
qlua:/home/ubuntu/Torch7/install/share/lua/5.1/nn/Container.lua:67:
在nn.Sequential的1个模块中:
/home/ubuntu/Torch7/install/share/lua/5.1/nn/THNN.lua:109:v的第3个参数无效(无法将“struct THCudaTensor *”转换为“struct THFloatTensor *”)
堆栈回溯:
[C]:在函数'v'中
/home/ubuntu/Torch7/install/share/lua/5.1/nn/THNN.lua:109:在函数'SpatialConvolutionMM_updateOutput'中
...Torch7/install/share/lua/5.1/nn/SpatialConvolutionMM.lua:79:在函数<_0 ... Torch7/install/share/lua/5.1/nn/SpatialConvolutionMM.lua:69>
[C]:在函数'xpcall'中
/home/ubuntu/Torch7/install/share/lua/5.1/nn/Container.lua:63: 在函数'rethrowErrors'中
/home/ubuntu/Torch7/install/share/lua/5.1/nn/Sequential.lua:44: 在函数'forward'中
train-on-mnist.lua:227: 在函数'opfunc'中
/home/ubuntu/Torch7/install/share/lua/5.1/optim/sgd.lua:44: 在函数'sgd'中
train-on-mnist.lua:280: 在函数'train'中
train-on-mnist.lua:383: 在主块中
警告:如果您下面看到一个堆栈跟踪,它不指向此错误发生的地方。请仅使用上面的一个。
堆栈回溯:
[C]:在0xb6ae7bdd处
[C]:在函数'error'中
/home/ubuntu/Torch7/install/share/lua/5.1/nn/Container.lua:67: 在函数'rethrowErrors'中
/home/ubuntu/Torch7/install/share/lua/5.1/nn/Sequential.lua:44: 在函数'forward'中 rain-on-mnist.lua:227: 在函数'opfunc'中
/home/ubuntu/Torch7/install/share/lua/5.1/optim/sgd.lua:44: 在函数'sgd'中
train-on-mnist.lua:280: 在函数'train'中
train-on-mnist.lua:383: 在主块中
编辑: 解决方案是通过使用“:cuda()”将数据,标签和标准转换为Cuda,然后脚本完美运行。谢谢@lejlot
点赞
评论区的留言会收到邮件通知哦~
推荐文章
- 如何将两个不同的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 代码?
- addEventListener 返回 nil Lua
- Lua中获取用户配置主目录的跨平台方法