1

jaxlibCUDA および cuDNN バージョンで動作する特定のバージョンをインストールしようとしています。READMEに従って、私は試しています

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'

これの原因やエラーを回避する方法を知っている人はいますか?

4

2 に答える 2

0

もう 1 つの回避策は、最初に特定のバージョンを選択してから、利用可能な wheel ファイルから目的のバージョンをインストールすることです。

$ pip install https://storage.googleapis.com/jax-releases/cuda11/jaxlib-0.1.76+cuda11.cudnn82-cp39-none-manylinux2010_x86_64.whl
于 2022-01-28T20:35:32.317 に答える