No module named 'kernelconv2d_cuda' about stfan HOT 4 CLOSED

Hello, thank you for providing the code of the paper. when I tried to run this code. I run the install.sh to install the module named 'kernelconv2d_cuda'. However, I got the following error: Can you give me some help? Thanks!

luded from KernelConv2D_cuda.cpp:4:0:
KernelConv2D_kernel.h:10:2: error: 'cudaStream_t' has not been declared
cudaStream_t stream
^
KernelConv2D_kernel.h:20:5: error: 'cudaStream_t' has not been declared
cudaStream_t stream
^
KernelConv2D_cuda.cpp: In function 'int KernelConv2D_forward_cuda(at::Tensor&, at::Tensor&, int, at::Tensor&)':
KernelConv2D_cuda.cpp:18:29: error: 'class at::Context' has no member named 'getCurrentCUDAStream'
at::globalContext().getCurrentCUDAStream()
^
KernelConv2D_cuda.cpp: In function 'int KernelConv2D_backward_cuda(at::Tensor&, at::Tensor&, int, at::Tensor&, at::Tensor&, at::Tensor&)':
KernelConv2D_cuda.cpp:42:29: error: 'class at::Context' has no member named 'getCurrentCUDAStream'
at::globalContext().getCurrentCUDAStream()
^
error: command 'x86_64-linux-gnu-gcc' failed with exit status 1

from stfan.

I solved this problem by removing my previous virtual enviroment completely and creating a new one. I reinstalled all packages required via 'pip' instead of 'conda' as suggested by the author. Hope that this will work for others as well. Note that torchvision==0.2.1 should be used for torch==0.4.1.
Thanks!

from stfan.

Thanks!

from stfan.

将setup.py中的 c++11改成c++14，就可以啦~

from stfan.