Files
MX_Pmodule/careless/build
T

28 lines
863 B
Python
Executable File

#!/usr/bin/env modbuild
pbuild::prep() {
:
}
pbuild::configure() {
# careless is pip-installed, but upstream recommends a fresh conda env
# (TensorFlow dependency conflicts), so install a private miniconda in PREFIX
mkdir -p "$PREFIX/miniconda"
wget https://repo.anaconda.com/miniconda/Miniconda3-latest-Linux-x86_64.sh -O "$PREFIX/miniconda/miniconda.sh"
bash "$PREFIX/miniconda/miniconda.sh" -b -u -p "$PREFIX/miniconda/"
"$PREFIX/miniconda/condabin/conda" create -y -n "careless_$V" python=3.12
}
pbuild::compile() {
:
}
pbuild::install() {
source "$PREFIX/miniconda/etc/profile.d/conda.sh"
conda activate "careless_$V"
pip install --upgrade pip
# [cuda] extra ships the NVIDIA libs via pip wheels; works on CPU-only nodes too
pip install --no-cache-dir "careless[cuda]==${V_PKG}"
conda deactivate
}