如何确定deepjava库（DJL）中可用的GPU内存？

有没有办法在DJL中查询GPU上的最大总内存？例如，NVIDIA 1080ti应返回11GB的值

---

我正在使用mxnet引擎，但引擎无关的解决方案当然更可取。

是的，DJL在年由JNA直接与CUDA API集成。所以这是引擎不可知的解决方案

MemoryUsage mem = CudaUtils.getGpuMemory(device);
mem.getMax(); // it should return 11GB