Установка jax и jaxlib для CUDA V9.1.85 ⇐ Python

[Anonymous]

Я использую 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 |
+-----------------------------------------------------------------------------+

Я пробовал установить JAX и jaxlib с помощью стандартных команд pip install jax и pip install -U "jax[cuda9]", но не могу заставить их работать.
Я пробовал установить JAX и jaxlib, используя стандартные команды pip install jax и pip install -U "jax[cuda9]", но не могу заставить их работать.
p>
Я тоже пробовал...
pip install jaxlib==0.1.69+cuda91 -f https://storage.googleapis.com/jax-rele ... eases.html
Looking in links: https://storage.googleapis.com/jax-rele ... eases.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


Подробнее здесь: https://stackoverflow.com/questions/791 ... da-v9-1-85