问题描述
我正在尝试安装特定版本的 jaxlib
以使用我的 CUDA 和 cuDNN 版本。按照自述文件,我正在尝试
pip install --upgrade jax jaxlib==0.1.52+cuda101 -f https://storage.googleapis.com/jax-releases/jax_releases.html
这将返回以下错误:
ERROR: Requested jaxlib==0.1.52+cuda101 from https://storage.googleapis.com/jax-releases/cuda101/jaxlib-0.1.52%2Bcuda101-cp37-none-manylinux2010_x86_64.whl has different version in Metadata: '0.1.52'
解决方法
此错误似乎来自 pip 版本 20.3.X 及更高版本中的新检查,可能与新的依赖项解析器有关。我可以使用 pip 版本 20.3.3 重现此错误,但使用 pip 版本 20.2.4 可以正确安装包。
最简单的方法可能是先降级 pip;即
pip install pip==20.2.4
然后继续安装 jaxlib。