Ubuntu18+Python3.8安装jaxlib==0.1.69+cuda118失败求助

错误原因 #

你指定的jaxlib==0.1.69+cuda118不存在，因为jaxlib 0.1.69版本发布时，CUDA 11.8还未被纳入支持范围，该版本仅提供适配CUDA 10.2、11.1等旧版本的包，所以pip找不到对应分发版本。

解决方案 #

根据你的CUDA版本选择以下方案：

方案1：使用jaxlib 0.1.69适配的CUDA版本 #

如果你安装的是CUDA 11.1，执行：

pip install --upgrade jax jaxlib==0.1.69+cuda111 -f https://storage.googleapis.com/jax-releases/jax_releases.html

如果是CUDA 10.2，执行：

pip install --upgrade jax jaxlib==0.1.69+cuda102 -f https://storage.googleapis.com/jax-releases/jax_releases.html

方案2：升级jax/jaxlib到支持CUDA 11.8的版本 #

如果你确实使用CUDA 11.8，需要升级到支持该版本的jax/jaxlib（同时保证和snerg项目兼容），推荐使用jaxlib 0.3.25（适配Python3.8且支持CUDA11.8），执行：

pip install --upgrade jax jaxlib==0.3.25+cuda118 -f https://storage.googleapis.com/jax-releases/jax_releases.html

安装完成后测试snerg运行情况，若出现兼容性问题，可尝试调整到其他支持CUDA11.8且兼容Python3.8的jaxlib版本（比如0.3.20、0.3.22等）。

额外注意 #

报错中提到的0.4.14 Requires-Python >=3.9是无关信息，只是pip自动过滤了不兼容Python3.8的新版本，不影响当前安装操作。

内容的提问来源于stack exchange，提问作者m0j1