如何安装带有GPU支持的objax？
我遵循objax文档来安装带有GPU支持的库：setup.html
即
pip install --upgrade objax
CUDA_VERSION=11.6
pip install -f https://storage.googleapis.com/jax-releases/jax_releases.html jaxlib==`python3 -c 'import jaxlib; print(jaxlib.__version__)'`+cuda`echo $CUDA_VERSION | sed s:\\\.::g`
然而，最后一步不起作用。我收到以下错误消息：
错误:无法找到满足jaxlib==0.3.15+cuda116要求的版本(来自版本: 0.1.32、0.1.40、0.1.41、0.1.42、0.1.43、0.1.44、0.1.46、0.1.50、0.1.51、0.1.52、0.1.55、0.1.56、0.1.57、0.1.58、0.1.59、0.1.60、0.1.61、0.1.62，0.1.63、0.1.64、0.1.65、0.1.66、0.1.67、0.1.68、0.1.69、0.1.70、0.1.71、0.1.72、0.1.73、0.1.74、0.1.75、0.1.76、0.3.0、0.3.2、0.3.5、0.3.7、0.3.8、0.3.10、0.3.14，0.3.15)错误:没有为jaxlib==0.3.15+cuda116找到匹配的分布
我尝试过使用python/CUDA的多个版本，但我总是遇到这个错误。
在开始时执行pip install --upgrade pip是没有帮助的。
系统说明：
- 操作系统: Ubuntu 20.04.4 LTS
- CUDA版本: 11.6
- Python版本: 3.8.13
回答 2
Stack Overflow用户
发布于 2022-08-24 18:37:55
JAX最近更新了它的GPU安装说明，您可以在这里找到：https://github.com/google/jax#pip-installation-gpu-cuda
特别是，数据自动化系统的车轮现在位于releases.html
因此，例如，您可以使用
$ pip install "jax[cuda11_cudnn805]" -f https://storage.googleapis.com/jax-releases/jax_cuda_releases.html并将cuda11和cudnn805分别替换为适合您的系统的CUDA和CUDNN版本，确保它们与上述URL索引中列出的版本相匹配。
我向objax存储库发送了一个拉请求，以更新您所遵循的说明：https://github.com/google/objax/pull/246
Stack Overflow用户
发布于 2022-08-24 12:37:59
文档页是2年前的最后更新。它已经过时了，正因为如此，它现在还在说谎。仔细看一看releases.html --目前还没有CUDA的版本。
至于非数据自动化系统版本，您可以直接从PyPI安装它们。
pip install jaxlib对于CUDA版本-试着编译从来源。或者使用conda：https://anaconda.org/search?q=jaxlib
https://stackoverflow.com/questions/73472417
复制相似问题
腾讯云开发者
