tensorflow / tensorflow/java

Can't use GPU on latest snapshot

未关闭
#384 6 条评论 0 个 reaction 已指派 0 人 在 GitHub 查看

还没有人认领这个 Issue。

主要语言
Java
星标
928
派生
227
PR 合并指标
30 天内没有已合并 PR

描述

On Windows, I can't use a GPU device even when depending on tensorflow-core-platform-gpu, and I don't see the dll finding output. Things work fine on 0.3.3, but not on 0.4.0-SNAPSHOT so I suspect it's related to the 2.6.0 upgrade.

贡献指南

打开贡献指南

从这里开始

  1. 先读完整个 Issue,再读项目的贡献指南。
  2. 在 Issue 下留言说明你要接手 —— 这能避免两个人做同样的事。
  3. Fork 仓库,在一个分支上完成修改。
  4. 提交 Pull Request,并在描述里引用这个 Issue 编号。

调研方向

首先比较 0.3.3 和 0.4.0-SNAPSHOT 之间的 Windows GPU 依赖项和原生 DLL 加载行为,尤其关注升级到 TensorFlow 2.6.0 的相关变化。使用 tensorflow-core-platform-gpu 重现该问题,并验证 GPU 使用正常,且最新 snapshot 中提供 DLL 查找输出。

由索引模型根据 Issue 内容生成。

评估

技术栈
java, tensorflow
领域
machine-learning, operating-systems
Issue 类型
缺陷
难度
4/5
预计耗时
3-5 天
活跃度
停滞
描述清晰度
需要澄清
新手友好度
30/100

把新 issue 发到你的邮箱

精选适合新手参与的 GitHub issue 摘要。