报错信息:
Collecting cached_property
Downloading http://mirrors.aliyun.com/pypi/packages/48/19/f2090f7dad41e225c7f2326e4cfe6fff49e57dedb5b53636c9551f86b069/cached_property-1.5.2-py2.py3-none-any.whl (7.6 kB)
ERROR: Could not find a version that satisfies the requirement jaxlib (from orbax-checkpoint->flax) (from versions: none)
ERROR: No matching distribution found for jaxlib (from orbax-checkpoint->flax)
解决方式:
安装jax和jaxlib库
pip install jax==0.3.13 https://whls.blob.core.windows.net/unstable/cuda111/jaxlib-0.3.7+cuda11.cudnn82-cp38-none-win_amd64.whl