File size: 956 Bytes
7934b29 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 |
#!/usr/bin/env bash
set -e
INSTALL_OPTION=${1:-"dev"}
PIP=pip
echo 'Uninstalling stuff'
${PIP} uninstall -y nemo_toolkit
${PIP} uninstall -y sacrebleu
# Kept for legacy purposes
${PIP} uninstall -y nemo_asr
${PIP} uninstall -y nemo_nlp
${PIP} uninstall -y nemo_tts
${PIP} install -U setuptools
if [ -n "${NVIDIA_PYTORCH_VERSION}" ]; then
echo 'Installing NeMo in NVIDIA PyTorch container:' "${NVIDIA_PYTORCH_VERSION}" 'so will not install numba'
else
if [ -n "${CONDA_PREFIX}" ]; then
NUMBA_VERSION=0.55
echo 'Installing numba=='${NUMBA_VERSION}
conda install -y -c conda-forge numba==${NUMBA_VERSION}
fi
fi
echo 'Installing nemo'
if [[ "$INSTALL_OPTION" == "dev" ]]; then
${PIP} install --editable ".[all]"
else
rm -rf dist/
${PIP} install build pytest-runner
python -m build --no-isolation --wheel
DIST_FILE=$(find ./dist -name "*.whl" | head -n 1)
${PIP} install "${DIST_FILE}[all]"
fi
echo 'All done!'
|