Advertisement

Torch7入门 第六版 使用多GPU的程序 （2018年4月18日更新）

阅读量：

总说

更新: 重写多GPU程序示例, 2018-4-18
不禁感慨一番，没想到时隔一年之后，我竟然再次对这一内容进行修订，原本是不打算再做修改的，毕竟错误就错误了，根本无人关注。然而网友的留言指出其中存在一些问题，因此决定还是进行一次更新吧，不多说了，真得给自己点个赞。
Torch中实现多GPU程序的编写实际上并不复杂。

最基础的代码示例如下：

第一种方法(不推荐)

复制代码 
    require 'optim'
    require 'nn'
    require 'cutorch'
    require 'cunn'
    require 'torch'
    torch.setdefaulttensortype('torch.FloatTensor')
    
    local conv_net = nn.Sequential()
    local concat = nn.ConcatTable()
    concat:add(nn.SpatialConvolution(3,3,3,3,1,1,1,1):noBias())
    concat:add(nn.SpatialConvolution(3,3,3,3,1,1,1,1):noBias())
    conv_net:add(concat)
    lo

全部评论 (0)

还没有任何评论哟~

相关文章推荐