Keras与TensorFlow/ PyTorch/ JAX 版本对应

截至2024/3/18

1.Keras + TensorFlow

1.1 Keras 2:

    tensorflow~=2.13.0 & keras~=2.13.0
    tensorflow~=2.14.0 & keras~=2.14.0
    tensorflow~=2.15.0 & keras~=2.15.0

1.2 Keras 3:

    tensorflow~=2.15.0 & keras~=3.0.0
    tensorflow~=2.16.1 & keras~=3.0.0

2.Keras + PyTorch

torch~=2.1.0 & keras~=3.0.0

3.Keras + JAX v

    jax==0.4.20 & keras~=3.0.0

   
 

你可能感兴趣的:(keras,人工智能,深度学习)