You need to enable JavaScript to run this app.
优惠活动
大模型
产品
解决方案
定价
更多

DJL无法调用已安装GPU的问题求助

DJL无法调用已安装GPU的问题求助

我最近测试DJL的时候遇上了个棘手的问题:不管是在Jupyter Notebook(Python环境)还是Java项目里,用MXNet和TensorFlow作为后端引擎,DJL都没法调用到GPU——所有测试场景下它都只用到CPU,这可把我愁坏了。

先跟大家说下我的环境配置:

  • Java环境:JDK 11
  • Python版本:13.3
  • Pip版本:24.3.1

我用的Java测试代码

Engine.debugEnvironment();
System.out.println(Engine.getAllEngines());
for (Device d : Engine.getInstance().getDevices())
    System.out.println(d.toString());
System.out.println(Engine.getInstance().getGpuCount());
NDManager manager = NDManager.newBaseManager();
var x = manager.arange(12);
System.out.println(x.getDevice());

项目依赖配置

<dependencies>
<dependency>
    <groupId>ai.djl</groupId>
    <artifactId>api</artifactId>
    <version>0.31.0</version>
</dependency>
<dependency>
    <...

有没有大佬能帮我排查下问题呀?到底是哪里配置漏了或者出错了,才导致DJL识别不到GPU呢?

备注:内容来源于stack exchange,提问作者ron

相关产品推荐
方舟 Agent Plan

超全模态模型 × Harness 升级,最新支持 Deepseek-V4.1-Flash、GLM-5.3 系列、Doubao-Seedream-5.0-pro、Kimi-K3 (部分), 限时 9.9 元起

最近更新时间:2026.04.15 11:23:04