- 博客(1)
- 收藏
- 关注
原创 JaxMARL配置踩坑记录
截止目前2025年4月,最新的jax和jaxlib已经更新到了0.5.3,cuda版本12。而JaxMARL最近一次更新是2024年年底,测试的版本是0.4.25,cuda版本11同时,不知道为什么,通过官方提供的gpu版本的安装指令无法安装gpu版本的jaxlib,因此会导致实际运行还是在cpu上运行.
2025-04-08 13:34:22
1205
空空如也
空空如也
TA创建的收藏夹 TA关注的收藏夹
TA关注的人
RSS订阅