Установите пакет Python с помощью pip, соблюдая текущую установку. ⇐ Python

[Anonymous]

Я хочу использовать контейнер сингулярности с предустановленным jax, например. на основе этого изображения. Кроме того, я хочу установить PyMC с numpyro, чтобы получить выборку NUTS с ускорением графического процессора.
Моя попытка — использовать файл def

Код: Выделить всё

Bootstrap: docker
From: nvcr.io/nvidia/jax:23.04-py3

%post
pip install pymc numpyro

%test
python -c "import jax;print(jax.devices())"

Тест не пройден, поскольку на последующем этапе локальные пакеты jax удаляются и устанавливается новая версия из PyPA. Это неприемлемо, поскольку образ докера поставляется с некоторыми очень специфическими локально скомпилированными версиями jaxlib и тому подобным.
Могу ли я дать несколько инструкций для pip, чтобы «установить эти два пакета и выбрать некоторые из них?» версию, совместимую с уже установленной"?

Подробнее здесь: https://stackoverflow.com/questions/792 ... stallation