-
Notifications
You must be signed in to change notification settings - Fork 14
Description
由于种种原因,无法安装旧版本的cupy,有什么方法可以解决compile_with_cache函数被删除的问题?
Traceback (most recent call last):
File "/data/sjh/DLGSANet-main/basicsr/archs/dlgsanet_arch.py", line 1025, in
_ = model(dummy_input)
File "/home/shnu_ibdi/anaconda3/envs/sjh/lib/python3.9/site-packages/torch/nn/modules/module.py", line 1194, in _call_impl
return forward_call(*input, **kwargs)
File "/data/sjh/DLGSANet-main/basicsr/archs/dlgsanet_arch.py", line 923, in forward
x = self.deep_feature_extraction(x) + x
File "/home/shnu_ibdi/anaconda3/envs/sjh/lib/python3.9/site-packages/torch/nn/modules/module.py", line 1194, in _call_impl
return forward_call(*input, **kwargs)
File "/home/shnu_ibdi/anaconda3/envs/sjh/lib/python3.9/site-packages/torch/nn/modules/container.py", line 204, in forward
input = module(input)
File "/home/shnu_ibdi/anaconda3/envs/sjh/lib/python3.9/site-packages/torch/nn/modules/module.py", line 1194, in _call_impl
return forward_call(*input, **kwargs)
File "/data/sjh/DLGSANet-main/basicsr/archs/dlgsanet_arch.py", line 708, in forward
return self.body(x) + x # shortcut in buildblock
File "/home/shnu_ibdi/anaconda3/envs/sjh/lib/python3.9/site-packages/torch/nn/modules/module.py", line 1194, in _call_impl
return forward_call(*input, **kwargs)
File "/home/shnu_ibdi/anaconda3/envs/sjh/lib/python3.9/site-packages/torch/nn/modules/container.py", line 204, in forward
input = module(input)
File "/home/shnu_ibdi/anaconda3/envs/sjh/lib/python3.9/site-packages/torch/nn/modules/module.py", line 1194, in _call_impl
return forward_call(*input, **kwargs)
File "/data/sjh/DLGSANet-main/basicsr/archs/dlgsanet_arch.py", line 562, in forward
x = self.IDynamicDWConv(self.norm1(x)) + x
File "/home/shnu_ibdi/anaconda3/envs/sjh/lib/python3.9/site-packages/torch/nn/modules/module.py", line 1194, in _call_impl
return forward_call(*input, **kwargs)
File "/data/sjh/DLGSANet-main/basicsr/archs/dlgsanet_arch.py", line 205, in forward
x = self.conv(x)
File "/home/shnu_ibdi/anaconda3/envs/sjh/lib/python3.9/site-packages/torch/nn/modules/module.py", line 1194, in _call_impl
return forward_call(*input, **kwargs)
File "/data/sjh/DLGSANet-main/basicsr/archs/dlgsanet_idynamicdwconv_util.py", line 307, in forward
out = _idynamic_cuda(x, weight, stride=1, padding=(self.kernel_size - 1) // 2)
File "/data/sjh/DLGSANet-main/basicsr/archs/dlgsanet_idynamicdwconv_util.py", line 255, in _idynamic_cuda
out = _idynamic.apply(input, weight, _pair(stride), _pair(padding), _pair(dilation))
File "/data/sjh/DLGSANet-main/basicsr/archs/dlgsanet_idynamicdwconv_util.py", line 178, in forward
f = load_kernel('idynamic_forward_kernel', _idynamic_kernel, Dtype=Dtype(input), nthreads=n,
File "cupy/_util.pyx", line 64, in cupy._util.memoize.decorator.ret
File "/data/sjh/DLGSANet-main/basicsr/archs/dlgsanet_idynamicdwconv_util.py", line 40, in load_kernel
kernel_code = cupy.cuda.compile_with_cache(code)
File "/home/shnu_ibdi/anaconda3/envs/sjh/lib/python3.9/site-packages/cupy/cuda/init.py", line 84, in getattr
raise AttributeError(
AttributeError: module 'cupy.cuda' has no attribute 'compile_with_cache'