为 CUDA V9.1.85 安装 jax 和 jaxlib

问题描述 投票：0回答：1

我正在使用Python 3.8.0和CUDA 9.1.85（CUDA编译工具，版本9.1）（或不使用？见下文）。不幸的是，我没有管理权限，所以我无法升级其中任何一个。

我在使用这些版本安装 JAX 和 jaxlib 时遇到问题。我该如何继续？

nvcc --version
nvcc: NVIDIA (R) Cuda compiler driver
Copyright (c) 2005-2017 NVIDIA Corporation
Built on Fri_Nov__3_21:07:56_CDT_2017
Cuda compilation tools, release 9.1, V9.1.85

nvidia-smi
Fri Nov 15 13:22:26 2024       
+-----------------------------------------------------------------------------+
| NVIDIA-SMI 470.223.02   Driver Version: 470.223.02   CUDA Version: 11.4     |
|-------------------------------+----------------------+----------------------+
| GPU  Name        Persistence-M| Bus-Id        Disp.A | Volatile Uncorr. ECC |
| Fan  Temp  Perf  Pwr:Usage/Cap|         Memory-Usage | GPU-Util  Compute M. |
|                               |                      |               MIG M. |
|===============================+======================+======================|
|   0  NVIDIA TITAN RTX    Off  | 00000000:0A:00.0 Off |                  N/A |
| 41%   31C    P8    18W / 280W |      8MiB / 24220MiB |      0%      Default |
|                               |                      |                  N/A |
+-------------------------------+----------------------+----------------------+
|   1  NVIDIA TITAN RTX    Off  | 00000000:42:00.0 Off |                  N/A |
| 41%   35C    P8    25W / 280W |     21MiB / 24219MiB |      0%      Default |
|                               |                      |                  N/A |
+-------------------------------+----------------------+----------------------+
                                                                               
+-----------------------------------------------------------------------------+
| Processes:                                                                  |
|  GPU   GI   CI        PID   Type   Process name                  GPU Memory |
|        ID   ID                                                   Usage      |
|=============================================================================|
|    0   N/A  N/A      1861      G   /usr/lib/xorg/Xorg                  4MiB |
|    1   N/A  N/A      1861      G   /usr/lib/xorg/Xorg                 16MiB |
+-----------------------------------------------------------------------------+

我尝试使用标准命令 pip install jax 和 pip install -U "jax[cuda9]" 安装 JAX 和 jaxlib，但我就是无法让它工作。

我也尝试过...

pip install jaxlib==0.1.69+cuda91 -f https://storage.googleapis.com/jax-releases/jax_cuda_releases.html
Looking in links: https://storage.googleapis.com/jax-releases/jax_cuda_releases.html
ERROR: Could not find a version that satisfies the requirement jaxlib==0.1.69+cuda91 (from versions: 0.1.52+cuda100, 0.1.52+cuda101, 0.1.52+cuda102, 0.1.52+cuda110, 0.1.55+cuda100, 0.1.55+cuda101, 0.1.55+cuda102, 0.1.55+cuda110, 0.1.56+cuda100, 0.1.56+cuda101, 0.1.56+cuda102, 0.1.56+cuda110, 0.1.56+cuda111, 0.1.57+cuda101, 0.1.57+cuda102, 0.1.57+cuda110, 0.1.57+cuda111, 0.1.58+cuda101, 0.1.58+cuda102, 0.1.58+cuda110, 0.1.58+cuda111, 0.1.59+cuda101, 0.1.59+cuda102, 0.1.59+cuda110, 0.1.59+cuda111, 0.1.60+cuda101, 0.1.60+cuda102, 0.1.60+cuda110, 0.1.60+cuda111, 0.1.61+cuda101, 0.1.61+cuda102, 0.1.61+cuda110, 0.1.61+cuda111, 0.1.61+cuda112, 0.1.62+cuda101, 0.1.62+cuda102, 0.1.62+cuda110, 0.1.62+cuda111, 0.1.62+cuda112, 0.1.63+cuda101, 0.1.63+cuda102, 0.1.63+cuda110, 0.1.63+cuda111, 0.1.63+cuda112, 0.1.64+cuda101, 0.1.64+cuda102, 0.1.64+cuda110, 0.1.64+cuda111, 0.1.64+cuda112, 0.1.65+cuda101, 0.1.65+cuda102, 0.1.65+cuda110, 0.1.65+cuda111, 0.1.65+cuda112, 0.1.66+cuda101, 0.1.66+cuda102, 0.1.66+cuda110, 0.1.66+cuda111, 0.1.67+cuda101, 0.1.67+cuda102, 0.1.67+cuda110, 0.1.67+cuda111, 0.1.68+cuda101, 0.1.68+cuda102, 0.1.68+cuda110, 0.1.68+cuda111, 0.1.69+cuda101, 0.1.69+cuda102, 0.1.69+cuda110, 0.1.69+cuda111, 0.1.70+cuda101, 0.1.70+cuda102, 0.1.70+cuda110, 0.1.70+cuda111, 0.1.71+cuda102, 0.1.71+cuda110, 0.1.71+cuda111, 0.1.72+cuda111, 0.1.73+cuda11.cudnn805, 0.1.73+cuda11.cudnn82, 0.1.74+cuda11.cudnn805, 0.1.74+cuda11.cudnn82, 0.1.75+cuda11.cudnn805, 0.1.75+cuda11.cudnn82, 0.1.76+cuda11.cudnn805, 0.1.76+cuda11.cudnn82, 0.3.0+cuda11.cudnn805, 0.3.0+cuda11.cudnn82, 0.3.2+cuda11.cudnn805, 0.3.2+cuda11.cudnn82, 0.3.5+cuda11.cudnn805, 0.3.5+cuda11.cudnn82, 0.3.7+cuda11.cudnn805, 0.3.7+cuda11.cudnn82, 0.3.8+cuda11.cudnn805, 0.3.8+cuda11.cudnn82, 0.3.10+cuda11.cudnn805, 0.3.10+cuda11.cudnn82, 0.3.14+cuda11.cudnn805, 0.3.14+cuda11.cudnn82, 0.3.15+cuda11.cudnn805, 0.3.15+cuda11.cudnn82, 0.3.18+cuda11.cudnn805, 0.3.18+cuda11.cudnn82, 0.3.20+cuda11.cudnn805, 0.3.20+cuda11.cudnn82, 0.3.22+cuda11.cudnn805, 0.3.22+cuda11.cudnn82, 0.3.24+cuda11.cudnn805, 0.3.24+cuda11.cudnn82, 0.3.25+cuda11.cudnn805, 0.3.25+cuda11.cudnn82, 0.4.1+cuda11.cudnn82, 0.4.1+cuda11.cudnn86, 0.4.2+cuda11.cudnn82, 0.4.2+cuda11.cudnn86, 0.4.3+cuda11.cudnn82, 0.4.3+cuda11.cudnn86, 0.4.4+cuda11.cudnn82, 0.4.4+cuda11.cudnn86, 0.4.6, 0.4.6+cuda11.cudnn82, 0.4.6+cuda11.cudnn86, 0.4.7, 0.4.7+cuda11.cudnn82, 0.4.7+cuda11.cudnn86, 0.4.7+cuda12.cudnn88, 0.4.9, 0.4.9+cuda11.cudnn86, 0.4.9+cuda12.cudnn88, 0.4.10, 0.4.10+cuda11.cudnn86, 0.4.10+cuda12.cudnn88, 0.4.11, 0.4.11+cuda11.cudnn86, 0.4.11+cuda12.cudnn88, 0.4.12, 0.4.12+cuda11.cudnn86, 0.4.12+cuda12.cudnn89, 0.4.13, 0.4.13+cuda11.cudnn86, 0.4.13+cuda12.cudnn89)
ERROR: No matching distribution found for jaxlib==0.1.69+cuda91
python cuda jax
1个回答
0
投票

CUDA 9 相当旧，较新版本的 JAX 不支持它（查看 JAX 变更日志，CUDA 9.2 支持在 2020 年 6 月发布的 版本 0.1.49 中被删除）。

您可以尝试安装 JAX v0.1.48，但我怀疑它是否适合您 例如，在发布时，Python 3.8 是最新版本，而 Python 3.8 现在已经已过其生命周期。

最好的选择是使用更新的 CUDA。如果您无法更新系统 CUDA，您始终可以按照JAX 安装中的说明使用 pip 安装的 CUDA：例如，这将从 PyPI 安装带有 CUDA12 轮子的最新 JAX：

pip install --upgrade "jax[cuda12]"
最新问题
© www.soinside.com 2019 - 2024. All rights reserved.