File size: 880 Bytes
a244197 | 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 | #!/bin/bash
# env_setup.sh - 公共环境初始化脚本
# 用法: source templates/env_setup.sh <conda_env> <module1> [module2 ...]
#
# 参数:
# $1 - conda 环境名 (如 chem)
# $2 ... - 需要加载的 module 列表
CONDA_ENV="${1:?请指定 conda 环境名}"
shift
# 系统脚本 (bashrc, conda, module) 可能引用未初始化变量或返回非零退出码，
# 临时关闭 errexit 和 nounset 避免报错
set +eu
# 清理环境
module purge
source ~/.bashrc
# 激活 conda 环境
conda activate "$CONDA_ENV"
# 加载指定 modules
for mod in "$@"; do
module load "$mod"
done
# 恢复 errexit 和 nounset
set -eu
# 加载数据集路径
SCRIPT_DIR="$(cd "$(dirname "${BASH_SOURCE[0]}")" && pwd)"
DEMO_DIR="$(dirname "$SCRIPT_DIR")"
REPO_ROOT="$(cd "$DEMO_DIR/../.." && pwd)"
if [ -f "$REPO_ROOT/env.sh" ]; then
source "$REPO_ROOT/env.sh"
fi
|