Train Network using GPU with customlayer

1 次查看（过去 30 天）
Rui Xiang
Rui Xiang 2018-4-19
评论： Rui Xiang 2018-5-1

Hi I wrote an custom layer which is about spherical CNN, but when I try it on GPU it returns this error messagee

 Error using trainNetwork (line 140)
Unexpected error calling cuDNN: CUDNN_STATUS_NOT_SUPPORTED.
 Caused by:
    Error using nnet.internal.cnngpu.reluBackward
    Unexpected error calling cuDNN: CUDNN_STATUS_NOT_SUPPORTED.

When I try CPU to train it, it works but very slow. How should I fix it? I can upload the code for my layer if needed.

Thanks:)

  3 个评论
显示 1更早的评论
Joss Knight
Joss Knight 2018-4-28
You would have to show us what your code is doing. Presumably you are passing invalid data out of layer and into the succeeding activation layer. You need to return a 4-D array of single precision values. Have you run checkLayer?
Rui Xiang
Rui Xiang 2018-5-1
It is 4D array, and it can be runned under cpu. And my matlab is 2017b so it doesn't have the checklayer function.

请先登录，再进行评论。

请先登录，再回答此问题。

回答（0 个）

类别

在 Help Center 和 File Exchange 中查找有关 Image Data Workflows 的更多信息

标签

产品

Community Treasure Hunt

Find the treasures in MATLAB Central and discover how the community can help you!

Start Hunting!

Translated by