emrahbasaran / SPReID

Code for our CVPR 2018 paper - Human Semantic Parsing for Person Re-identification
MIT License
141 stars 32 forks source link

cupy.cuda.compiler.CompileException: nvrtc: error: failed to load builtins #17

Closed prologueyan-hr closed 5 years ago

prologueyan-hr commented 5 years ago

First of all thanks your amazing work! when I run the code I have met the following issue: Traceback (most recent call last): File "main.py", line 300, in <module> Train() File "main.py", line 147, in Train Model[device_id](img, lbl, dataset, train=True) File "/home/wangliu/Hr/SPReID/modelx.py", line 387, in __call__ xo = self.segmentation.predictor(x) File "/home/wangliu/Hr/SPReID/modelx.py", line 324, in __call__ x = convolution(x) File "/home/wangliu/Hr/SPReID/modelx.py", line 315, in convolution x = self.convolution[0](x) File "/home/wangliu/Hr/SPReID/modelx.py", line 60, in __call__ x = self.conv(x) File "/home/wangliu/anaconda3/envs/aligned_reid/lib/python2.7/site-packages/chainer/link.py", line 242, in __call__ out = forward(*args, **kwargs) File "/home/wangliu/anaconda3/envs/aligned_reid/lib/python2.7/site-packages/chainer/links/connection/convolution_2d.py", line 173, in forward groups=self.groups) File "/home/wangliu/anaconda3/envs/aligned_reid/lib/python2.7/site-packages/chainer/functions/connection/convolution_2d.py", line 572, in convolution_2d y, = fnode.apply(args) File "/home/wangliu/anaconda3/envs/aligned_reid/lib/python2.7/site-packages/chainer/function_node.py", line 263, in apply outputs = self.forward(in_data) File "/home/wangliu/anaconda3/envs/aligned_reid/lib/python2.7/site-packages/chainer/function_node.py", line 369, in forward return self.forward_gpu(inputs) File "/home/wangliu/anaconda3/envs/aligned_reid/lib/python2.7/site-packages/chainer/functions/connection/convolution_2d.py", line 173, in forward_gpu return self._forward_gpu_core(x, W, b) File "/home/wangliu/anaconda3/envs/aligned_reid/lib/python2.7/site-packages/chainer/functions/connection/convolution_2d.py", line 180, in _forward_gpu_core cover_all=self.cover_all, dy=self.dy, dx=self.dx) File "/home/wangliu/anaconda3/envs/aligned_reid/lib/python2.7/site-packages/chainer/utils/conv.py", line 123, in im2col_gpu h, w, out_h, out_w, kh, kw, sy, sx, ph, pw, dy, dx, col) File "cupy/core/_kernel.pyx", line 558, in cupy.core._kernel.ElementwiseKernel.__call__ File "cupy/core/_kernel.pyx", line 579, in cupy.core._kernel.ElementwiseKernel._get_elementwise_kernel File "cupy/core/_kernel.pyx", line 392, in cupy.core._kernel._get_elementwise_kernel File "cupy/core/_kernel.pyx", line 26, in cupy.core._kernel._get_simple_elementwise_kernel File "cupy/core/_kernel.pyx", line 46, in cupy.core._kernel._get_simple_elementwise_kernel File "cupy/core/carray.pxi", line 148, in cupy.core.core.compile_with_cache File "/home/wangliu/anaconda3/envs/aligned_reid/lib/python2.7/site-packages/cupy/cuda/compiler.py", line 135, in compile_with_cache base = _preprocess('', options, arch) File "/home/wangliu/anaconda3/envs/aligned_reid/lib/python2.7/site-packages/cupy/cuda/compiler.py", line 98, in _preprocess result = prog.compile(options) File "/home/wangliu/anaconda3/envs/aligned_reid/lib/python2.7/site-packages/cupy/cuda/compiler.py", line 245, in compile raise CompileException(log, self.src, self.name, options) cupy.cuda.compiler.CompileException: nvrtc: error: failed to load builtins Do you know how to solve that?