((invalid argument at /pytorch/aten/src/THC/THCGeneral.cpp)) when runnning "python demo_MiddleBury.py"

[Rimay]

when running CUDA_VISIBLE_DEVICES=0 python demo_MiddleBury.py
I got the following error
THCudaCheck FAIL file=/pytorch/aten/src/THC/THCGeneral.cpp line=405 error=11 : invalid argument Traceback (most recent call last): File "demo_MiddleBury.py", line 131, in <module> y_s,offset,filter = model(torch.stack((X0, X1),dim = 0)) File "/data/anaconda3/lib/python3.7/site-packages/torch/nn/modules/module.py", line 489, in __call__ result = self.forward(*input, **kwargs) File "/data/DAIN/networks/DAIN.py", line 130, in forward cur_filter_input[:, 3:, ...]),dim=0)) File "/data/anaconda3/lib/python3.7/site-packages/torch/nn/modules/module.py", line 489, in __call__ result = self.forward(*input, **kwargs) File "/data/anaconda3/lib/python3.7/site-packages/torch/nn/modules/container.py", line 92, in forward input = module(input) File "/data/anaconda3/lib/python3.7/site-packages/torch/nn/modules/module.py", line 489, in __call__ result = self.forward(*input, **kwargs) File "/data/anaconda3/lib/python3.7/site-packages/torch/nn/modules/conv.py", line 320, in forward self.padding, self.dilation, self.groups) RuntimeError: cuda runtime error (11) : invalid argument at /pytorch/aten/src/THC/THCGeneral.cpp:405

PS: my environment is python3.7.3 + cuda9.0 + cudnn7.1.3 + pytorch1.0.0

[laomao0]

what is your compute version?
refer to web: https://en.wikipedia.org/wiki/CUDA
and issue: #19