[Bug]: build ditto时jax版本冲突

[zzx-QDU]

Issue Type

Build/Install

Modules Involved

Others

Have you reproduced the bug with SPU HEAD?

Yes

Have you searched existing issues?

Yes

SPU Version

ditto分支

OS Platform and Distribution

ubuntu22.04

Python Version

3.10

Compiler Version

No response

Current Behavior?

您好，我想学习下ditto的代码，但是我在编译时报了jax相关的问题。之后我按照网上的说法重新安装了较低版本的jax，结果又有很多其他版本冲突，请问如果想运行ditto，应该装什么版本的jax？

感谢帮助！

Standalone code to reproduce the issue

bazel run //spu/tests:jnp_dynamic_ring_test

Relevant log output

INFO: Analyzed target //spu/tests:jnp_dynamic_ring_test (0 packages loaded, 0 targets configured).
INFO: Found 1 target...
Target //spu/tests:jnp_dynamic_ring_test up-to-date:
  bazel-bin/spu/tests/jnp_dynamic_ring_test
INFO: Elapsed time: 0.392s, Critical Path: 0.00s
INFO: 1 process: 1 internal.
INFO: Build completed successfully, 1 total action
INFO: Running command line: bazel-bin/spu/tests/jnp_dynamic_ring_test
Traceback (most recent call last):
  File "/root/.cache/bazel/_bazel_root/c7c2256833a99c4ceaf0534f480b1c44/execroot/spulib/bazel-out/k8-fastbuild/bin/spu/tests/jnp_dynamic_ring_test.runfiles/spulib/spu/tests/jnp_dynamic_ring_test.py", line 22, in <module>
    import spu.utils.simulation as ppsim
  File "/root/.cache/bazel/_bazel_root/c7c2256833a99c4ceaf0534f480b1c44/execroot/spulib/bazel-out/k8-fastbuild/bin/spu/tests/jnp_dynamic_ring_test.runfiles/spulib/spu/utils/simulation.py", line 22, in <module>
    from jax import linear_util as jax_lu
ImportError: cannot import name 'linear_util' from 'jax' (/root/miniconda3/lib/python3.10/site-packages/jax/__init__.py)

[dz12123]

你好我也在学习ditto的代码，加个联系方式可以一起讨论吗

[zzx-QDU]

可以啊，怎么加您呢？

…

---- 回复的原邮件 ---- | 发件人 | ***@***.***> | | 日期 | 2025年02月11日 16:22 | | 收件人 | ***@***.***> | | 抄送至 | ***@***.***>***@***.***> | | 主题 | Re: [secretflow/spu] [Bug]: build ditto时jax版本冲突 (Issue #991) | 你好我也在学习ditto的代码，加个联系方式可以一起讨论吗 — Reply to this email directly, view it on GitHub, or unsubscribe. You are receiving this because you authored the thread.Message ID: ***@***.***>

[dz12123]

可以啊，怎么加您呢？
…

加我qq 528074410

[w-gc]

可以参考主分支修改下：

[Candicepan]

你好我也在学习ditto的代码，加个联系方式可以一起讨论吗

Welcome to join the discussion on GitHub Discussions! Anyone following ditto can participate together.