Train Network using GPU with customlayer

2018 4 19
0 个回答
更新时间：2018 5 1
3 次查看（30 天）

0 个投票

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更早的评论 隐藏 1更早的评论

Rui Xiang
Rui Xiang 2018-4-19
Is it because of the relu layer? Should I custom a relulayer according to my data structure as well?
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 个）

类别

在 帮助中心 和 File Exchange 中查找有关 Deep Learning Toolbox 的更多信息

产品

标签

Community Treasure Hunt

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

Start Hunting!

Translated by