Open2
Internal issue: Candidate is not for this requirement jax[cuda12-pip,cuda12-pip] vs jax[cuda12-pip]
pyproject.toml
[[tool.rye.sources]]
name = "google-jax"
url = "https://storage.googleapis.com/jax-releases/jax_cuda_releases.html"
type = "find-links"
rye add "jax[cuda12_pip]"
してrye sync
すると失敗する。
PyPIからSourceを落としてきて依存関係を確認する。
以下はjax-0.4.22.tar.gz
の場合の例。
jax.egg-info/requires.txt
ml_dtypes>=0.2.0
numpy>=1.22
opt_einsum
scipy>=1.9
[:python_version < "3.10"]
importlib_metadata>=4.6
[:python_version >= "3.11"]
numpy>=1.23.2
[:python_version >= "3.12"]
numpy>=1.26.0
scipy>=1.11.1
[australis]
protobuf<4,>=3.13
[ci]
jaxlib==0.4.20
[cpu]
jaxlib==0.4.21
[cuda]
jaxlib==0.4.21+cuda11.cudnn86
[cuda11_cudnn86]
jaxlib==0.4.21+cuda11.cudnn86
[cuda11_local]
jaxlib==0.4.21+cuda11.cudnn86
[cuda11_pip]
jaxlib==0.4.21+cuda11.cudnn86
nvidia-cublas-cu11>=11.11
nvidia-cuda-cupti-cu11>=11.8
nvidia-cuda-nvcc-cu11>=11.8
nvidia-cuda-runtime-cu11>=11.8
nvidia-cudnn-cu11>=8.8
nvidia-cufft-cu11>=10.9
nvidia-cusolver-cu11>=11.4
nvidia-cusparse-cu11>=11.7
nvidia-nccl-cu11>=2.18.3
[cuda12_local]
jaxlib==0.4.21+cuda12.cudnn89
[cuda12_pip]
jaxlib==0.4.21+cuda12.cudnn89
nvidia-cublas-cu12>=12.2.5.6
nvidia-cuda-cupti-cu12>=12.2.142
nvidia-cuda-nvcc-cu12>=12.2.140
nvidia-cuda-runtime-cu12>=12.2.140
nvidia-cudnn-cu12>=8.9
nvidia-cufft-cu12>=11.0.8.103
nvidia-cusolver-cu12>=11.5.2
nvidia-cusparse-cu12>=12.1.2.141
nvidia-nccl-cu12>=2.18.3
nvidia-nvjitlink-cu12>=12.2
[minimum-jaxlib]
jaxlib==0.4.19
[tpu]
jaxlib==0.4.21
libtpu-nightly==0.1.dev20231204
requests