[jax]jax判断GPU是否可用

云未归来

发布于 2025-07-20 12:01:54

7600

代码可运行

运行总次数：0

代码可运行

jax和jaxlib是一起的，所以我们可以通过jax或者jaxlib去判断GPU是否用。

jax判断：

import jax

print(jax.devices())

jaxlib判断：

from jax.lib import xla_bridge
print(xla_bridge.get_backend().platform)

本文参与腾讯云自媒体同步曝光计划，分享自作者个人站点/博客。

原始发表：2025-07-17，如有侵权请联系 cloudcommunity@tencent.com 删除

本文分享自作者个人站点/博客前往查看

如有侵权，请联系 cloudcommunity@tencent.com 删除。

本文参与腾讯云自媒体同步曝光计划，欢迎热爱写作的你一起参与！

登录后参与评论

0 条评论

热度