报错信息:
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
在尝试安装flax库时遇到一个错误,报错信息指出找不到满足要求的jaxlib版本。为了解决这个问题,文章提供了安装jax和特定版本jaxlib(0.3.7+cuda11.cudnn82)的命令,这个版本适用于CUDA11.1和AMD64架构的系统。
7959

被折叠的 条评论
为什么被折叠?



