环境:ubuntu18.04,conda4.7.12,python3.6.9,tensorflow-gpu1.14,编译器pycharm。
训练3dcnn,在优化步骤报错no algorithm worked。
一开始还以为是tensorflow-gpu不支持3dcnn,因为同样程序在tensorflow-cpu版可以正常训练,而且报的错误也不是OOM。
后来发现就是因为显存不够,降低batchsize即可
本文发布于:2024-02-02 18:54:00,感谢您对本站的认可!
本文链接:https://www.4u4v.net/it/170687123845763.html
版权声明:本站内容均来自互联网,仅供演示用,请勿用于商业和其他非法用途。如果侵犯了您的权益请与我们联系,我们将在24小时内删除。
留言与评论(共有 0 条评论) |