search

SeitaroShinagawa / chainer-partial_convolution_image_inpainting

Reproduction of Nvidia image inpainting paper "Image Inpainting for Irregular Holes Using Partial Convolutions"
MIT License
114 stars 30 forks source link

CUBLAS_STATUS_NOT_SUPPORTED #22

Open chudonezen opened 6 years ago

chudonezen commented 6 years ago

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
SeitaroShinagawa commented 5 years ago

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)