RuntimeError: cuDNN error: CUDNN_STATUS_EXECUTION_FAILED

foamliu / Deep-Image-Matting-PyTorch

Deep Image Matting implementation in PyTorch

MIT License

820 stars 185 forks source link

RuntimeError: cuDNN error: CUDNN_STATUS_EXECUTION_FAILED #27

Closed huangjunxiong11 closed 4 years ago

huangjunxiong11 commented 4 years ago

When i run train.py, I get this error "RuntimeError: cuDNN error: CUDNN_STATUS_EXECUTION_FAILED", What made this error? What should I do? The configuration environment of my computer is ‘python3.5 pytorch1.1.0 cudn9.0 cudnn7.1.3 torchvision0.3.0’

foamliu commented 4 years ago

Seems an environment issue. Please make sure the environment setup (pytorch, cuda, cudnn) is right.

foamliu commented 4 years ago

Especially cuda and cudnn versions are compatiable.