【发布时间】:2020-11-12 18:30:13
【问题描述】:
我刚刚更新了我的服务器,并且正在使用 Deeplearning4j 运行 Web 服务。我想检查不同的性能配置,以决定未来的努力方向。据了解。
- 本机或 CPU,这是我当前的默认设置
- CUDA 或 GPU 使用情况。在代码中设置环境以使用 CUDA
- GTX GPU 专用。
对于#2,我认为除了在我的 Java 代码中设置环境之外,我还需要从类路径中删除 ND4J 本机 jar。
对于#3,我不确定这是否适用于我使用 DNN 模型但仅用于训练时?
非常感谢您的反馈!
【问题讨论】:
-
您好,这实际上取决于您的网络规模和问题类型。对此并没有真正的 1 个简单的答案。例如,较小的网络在 cpu 上速度较慢,而 gpus 需要批处理。如果您想获得更好的答案,则需要更具体。
-
谢谢亚当。我的模型非常小,文件大小为 3KB。我里面可能只有 4 层。一个 300x300 像素的图像切片有 300 个输入,图像大小可以高达 3k x 1k。最好的问候!
标签: deeplearning4j