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
相关产品推荐
相关产品推荐

