我坚持使用具有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上执行此操作的梦想?