chainer-partial_convolution_image_inpainting
chainer-partial_convolution_image_inpainting copied to clipboard
CUBLAS_STATUS_NOT_SUPPORTED
Hi ! Very thank you for your excellent job! I am trying to read the paper and run your code.I tools are
python3.6+chainer 4.2.0 +CUDA8.0
but it give me a problem "CUBLAS_STATUS_NOT_SUPPORTED", I tried find methods, but finde none (crying) could you give me some tips?
These are the problem details
File "/home/cxd/code/code_python/chainer-partial_convolution_image_inpainting-master/chainer-partial_convolution_image_inpainting-master/updater.py", line 133, in update_core
L_style = calc_loss_style(fs_I_out,fs_I_comp,fs_I_gt) #Loss style out and comp
File "/home/cxd/code/code_python/chainer-partial_convolution_image_inpainting-master/chainer-partial_convolution_image_inpainting-master/updater.py", line 48, in calc_loss_style
hout_gram = F.batch_matmul(hout,hout,transb=True)
File "/root/anaconda3/lib/python3.6/site-packages/chainer/functions/math/matmul.py", line 276, in batch_matmul
return BatchMatMul(transa=transa, transb=transb).apply((a, b))[0]
File "/root/anaconda3/lib/python3.6/site-packages/chainer/function_node.py", line 258, in apply
outputs = self.forward(in_data)
File "/root/anaconda3/lib/python3.6/site-packages/chainer/functions/math/matmul.py", line 210, in forward
return _batch_matmul(a, b, self.transa, self.transb, False),
File "/root/anaconda3/lib/python3.6/site-packages/chainer/functions/math/matmul.py", line 178, in _batch_matmul
return _matmul(a, b, transa, transb, transout)
File "/root/anaconda3/lib/python3.6/site-packages/chainer/functions/math/matmul.py", line 49, in _matmul
return xp.matmul(a, b)
File "cupy/core/core.pyx", line 3525, in cupy.core.core.matmul
File "cupy/core/core.pyx", line 3682, in cupy.core.core.matmul
File "cupy/cuda/cublas.pyx", line 700, in cupy.cuda.cublas.sgemmStridedBatched
File "cupy/cuda/cublas.pyx", line 718, in cupy.cuda.cublas.sgemmStridedBatched
File "cupy/cuda/cublas.pyx", line 267, in cupy.cuda.cublas.check_status
Sorry for the late response.
Have you already installed cupy (https://cupy.chainer.org/)?
Chainer depends on cupy for the gpu computing.
If not, you can install it by pip install cupy-cuda80
(for CUDA8.0)