pin numba=0.57.1 to fix reinstall.sh error (#7366)

Signed-off-by: Xuesong Yang <[email protected]>
NVIDIA · Sep 5, 2023 · eba7ab8 · eba7ab8
1 parent 888ef06
commit eba7ab8
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/reinstall.sh b/reinstall.sh
@@ -20,7 +20,7 @@ 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
+    NUMBA_VERSION=0.57.1
     echo 'Installing numba=='${NUMBA_VERSION}
     conda install -y -c conda-forge numba==${NUMBA_VERSION}
   fi