[ERGENT !!] cupy.cuda.driver.CUDADriverError: CUDA_ERROR_INVALID_PTX: a PTX JIT compilation failed

[jingcaiguo]

My code is:

import torch
from torch.autograd import Variable
import pyinn as P

x = Variable(torch.randn(4, 3, 3, 3).cuda())
kernel = 1
pad = 0
stride = (1, 1)

col_x = P.im2col(x, kernel, stride, pad)
img_x = P.col2im(col_x, kernel, stride, pad)

===============================================
BUT ENCOUNTER THIS ERROR:

Traceback (most recent call last):
File "/home/akuma/桌面/SR/BasicSR/codes/test_cupy.py", line 10, in
col_x = P.im2col(x, kernel, stride, pad)
File "/home/akuma/anaconda3/lib/python3.6/site-packages/pyinn/im2col.py", line 244, in im2col
return Im2Col(kernel_size, stride, padding)(input)
File "/home/akuma/anaconda3/lib/python3.6/site-packages/pyinn/im2col.py", line 207, in forward
return im2col_batch(input, self.kernel_size, self.stride, self.padding)
File "/home/akuma/anaconda3/lib/python3.6/site-packages/pyinn/im2col.py", line 183, in im2col_batch
_im2col(x, kernel_size, stride, padding, out=o)
File "/home/akuma/anaconda3/lib/python3.6/site-packages/pyinn/im2col.py", line 118, in _im2col
channels=n_input_plane)
File "cupy/util.pyx", line 48, in cupy.util.memoize.decorator.ret
File "/home/akuma/anaconda3/lib/python3.6/site-packages/pyinn/utils.py", line 20, in load_kernel
kernel_code = cupy.cuda.compile_with_cache(code)
File "/home/akuma/anaconda3/lib/python3.6/site-packages/cupy/cuda/compiler.py", line 166, in compile_with_cache
ls.add_ptr_data(ptx, six.u('cupy.ptx'))
File "cupy/cuda/function.pyx", line 205, in cupy.cuda.function.LinkState.add_ptr_data
File "cupy/cuda/function.pyx", line 207, in cupy.cuda.function.LinkState.add_ptr_data
File "cupy/cuda/driver.pyx", line 152, in cupy.cuda.driver.linkAddData
File "cupy/cuda/driver.pyx", line 81, in cupy.cuda.driver.check_status
cupy.cuda.driver.CUDADriverError: CUDA_ERROR_INVALID_PTX: a PTX JIT compilation failed

Process finished with exit code 1

=======================================
Anyone can solve this problem?