是否可以使用计算能力3.5 GPU对Torch中的阵列进行上采样?

时间:2016-04-10 21:33:48

标签: lua cuda torch

我坚持使用具有CUDA计算能力3.5而不是5的GPU,而且我一直在寻找一种方法来在Torch中进行空间上采样。 nn.SpatialUpSamplingNearest抛出错误,并且没有cuDNN替代方案。此外,我找不到任何方法来实现我自己的版本,因为看起来没有办法做一个不连续的任务......例如

th> a = torch.zeros(3,3):cuda()
th> a[{{1,2},{1,2}}]=1
cuda runtime error (8) : invalid device function at /tmp/luarocks_cutorch-scm-1-4917/cutorch/lib/THC/THCTensorMath.cu:25
stack traceback:
        [C]: at 0x7f84c1e410a0
        [C]: in function '__newindex'
        [string "a[{{1,2},{1,2}}]=1"]:1: in main chunk

有人知道如何让图像上采样工作,还是我应该放弃在GPU上执行此操作的梦想?

0 个答案:

没有答案