Я хочу использовать контейнер сингулярности с предустановленным 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
Мобильная версия