![](https://img-blog.csdnimg.cn/20201014180756757.png?x-oss-process=image/resize,m_fixed,h_64,w_64)
deep learing
zaqygv
这个作者很懒,什么都没留下…
展开
-
java tensorflow GPU环境运行的问题
问题:在本地cpu环境下运行良好,放到服务器上运行时报如下错误Could not create cudnn handle: CUDNN_STATUS_INTERNAL_ERRORis org.tensorflow.TensorFlowException: 2 root error(s) found. (0) Unknown: Failed to get convolution algorithm. This is probably because cuDNN failed to initial..原创 2021-09-24 13:43:54 · 161 阅读 · 0 评论 -
java tensorflow gpu内存使用太多的问题
平台linux,tensorflow版本1.15,gpu内存观察工具nvidim-smi问题:加载一个yolo2模型使用的gpu内存:在python中只有100M,在java中有10G以上。解决:创建会话时,配置GPU内存相关选项import org.tensorflow.*;import org.tensorflow.proto.framework.ConfigProto;import org.tensorflow.proto.framework.GPUOptions;Conf原创 2021-09-22 17:16:21 · 246 阅读 · 0 评论