2024年11月5日
Jax中关于Device ID的配置
摘要： Jax中关于Device ID的配置 本文主要介绍了2个在Jax框架中配置显卡Device ID的方法。第一种方法可以使用环境变量进行配置，对于众多的深度学习框架都是可以兼容的。而第二种方案是在Jax即时编译的过程中通过Jax生成的Device对象来控制数据的传输和函数执行的Device ID。 阅读全文
posted @ 2024-11-05 16:47 DECHIN 阅读(27) 评论(0) 推荐(0) 编辑