diff --git a/scripts/init.sh b/scripts/init.sh index d6e29d238ce34449e3d354575f8fa6e26cc2946c..5cb26dfcf2c3de16ad8b475eb438b959021062d5 100644 --- a/scripts/init.sh +++ b/scripts/init.sh @@ -11,12 +11,12 @@ CUDA_VERSION=11.6 # CUDA version (default: 11.6) # Load modules module load python3/$PYTHON_VERSION -module load $(module avail -o modulepath -t -C "python-${PYTHON_VERSION}" | grep "numpy/") -module load $(module avail -o modulepath -t -C "python-${PYTHON_VERSION}" | grep "scipy/") -module load $(module avail -o modulepath -t -C "python-${PYTHON_VERSION}" | grep "matplotlib/") -module load $(module avail -o modulepath -t -C "python-${PYTHON_VERSION}" | grep "pandas/") +module load $(module avail -o modulepath -t -C "python-${PYTHON_VERSION}" 2>&1 | grep "numpy/") +module load $(module avail -o modulepath -t -C "python-${PYTHON_VERSION}" 2>&1 | grep "scipy/") +module load $(module avail -o modulepath -t -C "python-${PYTHON_VERSION}" 2>&1 | grep "matplotlib/") +module load $(module avail -o modulepath -t -C "python-${PYTHON_VERSION}" 2>&1 | grep "pandas/") module load cuda/$CUDA_VERSION -CUDNN_MOD=$(module avail -o modulepath -t cudnn | grep "cuda-${CUDA_VERSION}" | sort | tail -n1) +CUDNN_MOD=$(module avail -o modulepath -t cudnn | grep "cuda-${CUDA_VERSION}" 2>&1 | sort | tail -n1) if [[ ${CUDNN_MOD} ]] then module load ${CUDNN_MOD}