454 lines
15 KiB
Bash
454 lines
15 KiB
Bash
#!/bin/bash
|
||
set -euo pipefail
|
||
|
||
# 定义颜色常量，用于输出提示
|
||
RED='\033[0;31m'
|
||
GREEN='\033[0;32m'
|
||
YELLOW='\033[1;33m'
|
||
NC='\033[0m' # No Color
|
||
|
||
# 全局变量 - 自动识别系统信息
|
||
OS_TYPE=""
|
||
ARCH_TYPE=""
|
||
CONDA_PATH="$HOME/mambaconda"
|
||
# 稳定版版本号（动态获取）
|
||
LATEST_CONDA_VERSION=""
|
||
LATEST_NVM_VERSION=""
|
||
|
||
# 打印信息函数
|
||
info() {
|
||
echo -e "${GREEN}[INFO]${NC} $1"
|
||
}
|
||
|
||
# 打印警告函数
|
||
warn() {
|
||
echo -e "${YELLOW}[WARN]${NC} $1"
|
||
}
|
||
|
||
# 打印错误函数
|
||
error() {
|
||
echo -e "${RED}[ERROR]${NC} $1"
|
||
exit 1
|
||
}
|
||
|
||
# 检查系统类型和架构
|
||
detect_system() {
|
||
info "开始检测系统信息..."
|
||
|
||
# 检测操作系统类型
|
||
if [[ "$OSTYPE" == "linux-gnu"* ]]; then
|
||
OS_TYPE="linux"
|
||
elif [[ "$OSTYPE" == "darwin"* ]]; then
|
||
OS_TYPE="macos"
|
||
else
|
||
error "不支持的操作系统：$OSTYPE，仅支持 Linux 和 macOS"
|
||
fi
|
||
|
||
# 检测系统架构
|
||
ARCH=$(uname -m)
|
||
if [[ "$ARCH" == "x86_64" || "$ARCH" == "amd64" ]]; then
|
||
ARCH_TYPE="x86_64"
|
||
elif [[ "$ARCH" == "aarch64" || "$ARCH" == "arm64" ]]; then
|
||
ARCH_TYPE="aarch64"
|
||
else
|
||
error "不支持的系统架构：$ARCH，仅支持 x86_64/amd64 和 aarch64/arm64"
|
||
fi
|
||
|
||
info "系统检测完成：$OS_TYPE $ARCH_TYPE"
|
||
}
|
||
|
||
# 获取最新稳定版版本号
|
||
get_stable_versions() {
|
||
info "开始获取各工具最新稳定版版本号..."
|
||
|
||
# 获取 Mambaconda 最新稳定版（跳过 pre-release）
|
||
info "获取 Mambaconda 最新稳定版..."
|
||
LATEST_CONDA_INFO=$(curl -s https://api.github.com/repos/conda-forge/miniforge/releases/latest)
|
||
LATEST_CONDA_VERSION=$(echo "$LATEST_CONDA_INFO" | grep -Po '"tag_name": "\K.*?(?=")')
|
||
if [ -z "$LATEST_CONDA_VERSION" ]; then
|
||
error "无法获取 Mambaconda 最新稳定版版本号，请检查网络"
|
||
fi
|
||
info "Mambaconda 最新稳定版：$LATEST_CONDA_VERSION"
|
||
|
||
# 获取 NVM 最新稳定版（跳过 pre-release）
|
||
info "获取 NVM 最新稳定版..."
|
||
LATEST_NVM_INFO=$(curl -s https://api.github.com/repos/nvm-sh/nvm/releases/latest)
|
||
LATEST_NVM_VERSION=$(echo "$LATEST_NVM_INFO" | grep -Po '"tag_name": "\K.*?(?=")')
|
||
if [ -z "$LATEST_NVM_VERSION" ]; then
|
||
# 备用：使用已知稳定版
|
||
LATEST_NVM_VERSION="v0.39.7"
|
||
warn "无法获取 NVM 最新稳定版，使用备用稳定版：$LATEST_NVM_VERSION"
|
||
fi
|
||
info "NVM 最新稳定版：$LATEST_NVM_VERSION"
|
||
}
|
||
|
||
# 检查是否为 root 用户（macOS 不建议 root）
|
||
check_root() {
|
||
if [ "$(id -u)" -eq 0 ]; then
|
||
if [[ "$OS_TYPE" == "macos" ]]; then
|
||
error "macOS 系统禁止使用 root 用户运行此脚本，请使用普通用户"
|
||
else
|
||
warn "当前为 root 用户，将安装到 /root/mambaconda 目录"
|
||
fi
|
||
fi
|
||
}
|
||
|
||
# 安装系统依赖工具
|
||
install_dependencies() {
|
||
info "开始安装基础依赖工具..."
|
||
|
||
if [[ "$OS_TYPE" == "linux" ]]; then
|
||
if command -v apt &> /dev/null; then
|
||
apt update && apt install -y curl wget git ca-certificates --no-install-recommends
|
||
elif command -v yum &> /dev/null; then
|
||
yum install -y curl wget git ca-certificates
|
||
elif command -v dnf &> /dev/null; then
|
||
dnf install -y curl wget git ca-certificates
|
||
else
|
||
error "Linux 系统不支持的包管理器，请手动安装 curl wget git"
|
||
fi
|
||
elif [[ "$OS_TYPE" == "macos" ]]; then
|
||
# 检查是否安装 brew
|
||
if ! command -v brew &> /dev/null; then
|
||
warn "未检测到 Homebrew，正在自动安装..."
|
||
/bin/bash -c "$(curl -fsSL https://raw.githubusercontent.com/Homebrew/install/HEAD/install.sh)"
|
||
fi
|
||
brew install curl wget git
|
||
fi
|
||
|
||
info "基础依赖工具安装完成"
|
||
}
|
||
|
||
# 安装 mambaconda 并配置清华源（仅稳定版）
|
||
install_mambaconda() {
|
||
info "开始安装 Mambaconda（稳定版 $LATEST_CONDA_VERSION）..."
|
||
# 检查是否已安装 conda
|
||
if command -v conda &> /dev/null; then
|
||
warn "检测到已安装 conda，跳过安装步骤"
|
||
configure_conda_env
|
||
return
|
||
fi
|
||
# Miniforge3-26.1.0-0-Linux-x86_64.sh
|
||
# 构建安装包名称
|
||
if [[ "$OS_TYPE" == "linux" ]]; then
|
||
MAMBAFORGE_PACKAGE="Miniforge3-${LATEST_CONDA_VERSION}-Linux-${ARCH_TYPE}.sh"
|
||
elif [[ "$OS_TYPE" == "macos" ]]; then
|
||
if [[ "$ARCH_TYPE" == "x86_64" ]]; then
|
||
MAMBAFORGE_PACKAGE="Miniforge3-${LATEST_CONDA_VERSION}-MacOSX-x86_64.sh"
|
||
else
|
||
MAMBAFORGE_PACKAGE="Miniforge3-${LATEST_CONDA_VERSION}-MacOSX-arm64.sh"
|
||
fi
|
||
fi
|
||
# https://github.com/conda-forge/miniforge/releases/download/26.1.0-0/Miniforge3-26.1.0-0-Linux-x86_64.sh
|
||
# 构建下载 URL（仅稳定版）
|
||
GITHUB_URL="https://github.com/conda-forge/miniforge/releases/download/${LATEST_CONDA_VERSION}/${MAMBAFORGE_PACKAGE}"
|
||
USTC_URL="https://mirrors.ustc.edu.cn/github-release/conda-forge/miniforge/${LATEST_CONDA_VERSION}/${MAMBAFORGE_PACKAGE}"
|
||
BACKUP_URL="https://mirrors.tuna.tsinghua.edu.cn/anaconda/miniconda/Miniforge3-Linux-${ARCH_TYPE}.sh"
|
||
|
||
# 下载安装包（使用中科大镜像源）
|
||
info "下载 Mambaconda 安装包（中科大源）..."
|
||
if ! wget -q "$USTC_URL" -O "/tmp/$MAMBAFORGE_PACKAGE"; then
|
||
info "中科大源下载失败，尝试 GitHub 官方源..."
|
||
if ! wget -q "$GITHUB_URL" -O "/tmp/$MAMBAFORGE_PACKAGE"; then
|
||
info "GitHub 源下载失败，尝试备用源..."
|
||
if ! wget -q "$BACKUP_URL" -O "/tmp/$MAMBAFORGE_PACKAGE"; then
|
||
error "所有源下载 mambaconda 安装包均失败，请手动下载稳定版安装包到 /tmp 目录后重新运行"
|
||
fi
|
||
fi
|
||
fi
|
||
|
||
# 静默安装（不修改 shell 配置文件）
|
||
bash "/tmp/$MAMBAFORGE_PACKAGE" -b -p "$CONDA_PATH" || error "mambaconda 安装失败"
|
||
|
||
# 删除安装包
|
||
rm -f "/tmp/$MAMBAFORGE_PACKAGE"
|
||
|
||
# 配置 conda 环境变量
|
||
configure_conda_env
|
||
|
||
# 激活 conda
|
||
source "$CONDA_PATH/etc/profile.d/conda.sh"
|
||
source "$CONDA_PATH/etc/profile.d/mamba.sh"
|
||
|
||
# 配置清华源
|
||
info "配置 conda 清华源..."
|
||
cat > "$HOME/.condarc" << EOF
|
||
channels:
|
||
- defaults
|
||
show_channel_urls: true
|
||
default_channels:
|
||
- https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main
|
||
- https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/r
|
||
- https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/msys2
|
||
custom_channels:
|
||
conda-forge: https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud
|
||
msys2: https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud
|
||
bioconda: https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud
|
||
menpo: https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud
|
||
pytorch: https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud
|
||
pytorch-lts: https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud
|
||
simpleitk: https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud
|
||
EOF
|
||
|
||
# 更新 conda 并安装 Python 3.10
|
||
info "安装 Python 3.10..."
|
||
conda install -y python=3.10 || error "Python 3.10 安装失败"
|
||
|
||
# 初始化 conda 到 shell
|
||
conda init bash
|
||
if command -v zsh &> /dev/null; then
|
||
conda init zsh
|
||
fi
|
||
if [[ "$OS_TYPE" == "macos" ]] && command -v fish &> /dev/null; then
|
||
conda init fish
|
||
fi
|
||
|
||
info "Mambaconda 稳定版安装并配置完成"
|
||
}
|
||
|
||
# 配置 conda 环境变量
|
||
configure_conda_env() {
|
||
info "配置 conda 环境变量..."
|
||
# 检查 .bashrc 中是否已有 conda 配置
|
||
if ! grep -q "conda initialize" "$HOME/.bashrc"; then
|
||
cat >> "$HOME/.bashrc" << EOF
|
||
|
||
# >>> conda initialize >>>
|
||
__conda_setup="\$('$CONDA_PATH/bin/conda' 'shell.bash' 'hook' 2> /dev/null)"
|
||
if [ \$? -eq 0 ]; then
|
||
eval "\$__conda_setup"
|
||
else
|
||
if [ -f "$CONDA_PATH/etc/profile.d/conda.sh" ]; then
|
||
. "$CONDA_PATH/etc/profile.d/conda.sh"
|
||
else
|
||
export PATH="$CONDA_PATH/bin:\$PATH"
|
||
fi
|
||
fi
|
||
unset __conda_setup
|
||
# <<< conda initialize <<<
|
||
EOF
|
||
fi
|
||
|
||
# 配置 zsh（如果存在）
|
||
if command -v zsh &> /dev/null && ! grep -q "conda initialize" "$HOME/.zshrc"; then
|
||
cat >> "$HOME/.zshrc" << EOF
|
||
|
||
# >>> conda initialize >>>
|
||
__conda_setup="\$('$CONDA_PATH/bin/conda' 'shell.zsh' 'hook' 2> /dev/null)"
|
||
if [ \$? -eq 0 ]; then
|
||
eval "\$__conda_setup"
|
||
else
|
||
if [ -f "$CONDA_PATH/etc/profile.d/conda.sh" ]; then
|
||
. "$CONDA_PATH/etc/profile.d/conda.sh"
|
||
else
|
||
export PATH="$CONDA_PATH/bin:\$PATH"
|
||
fi
|
||
fi
|
||
unset __conda_setup
|
||
# <<< conda initialize <<<
|
||
EOF
|
||
fi
|
||
|
||
# 立即生效环境变量
|
||
export PATH="$CONDA_PATH/bin:$PATH"
|
||
info "conda 环境变量配置完成"
|
||
}
|
||
|
||
# 安装 nvm 并配置淘宝源（仅稳定版）
|
||
install_nvm() {
|
||
info "开始安装 NVM（稳定版 $LATEST_NVM_VERSION）..."
|
||
# 检查是否已安装 nvm
|
||
if [ -d "$HOME/.nvm" ]; then
|
||
warn "检测到已安装 nvm，跳过安装步骤"
|
||
configure_nvm_env
|
||
return
|
||
fi
|
||
|
||
# 构建 nvm 安装脚本 URL（仅稳定版）
|
||
NVM_INSTALL_URL="https://raw.githubusercontent.com/nvm-sh/nvm/${LATEST_NVM_VERSION}/install.sh"
|
||
NVM_MIRROR_URL="https://cdn.jsdelivr.net/gh/nvm-sh/nvm@${LATEST_NVM_VERSION}/install.sh"
|
||
|
||
# 下载并安装 nvm（使用国内镜像源）
|
||
info "下载 NVM 安装脚本（稳定版）..."
|
||
if ! wget -q "$NVM_MIRROR_URL" -O "/tmp/install_nvm.sh"; then
|
||
info "镜像源下载失败，尝试 GitHub 官方源..."
|
||
if ! wget -q "$NVM_INSTALL_URL" -O "/tmp/install_nvm.sh"; then
|
||
error "下载 NVM 稳定版安装脚本失败，请检查网络"
|
||
fi
|
||
fi
|
||
|
||
bash "/tmp/install_nvm.sh" || error "nvm 安装失败"
|
||
|
||
# 删除安装包
|
||
rm -f "/tmp/install_nvm.sh"
|
||
|
||
# 配置 nvm 环境变量
|
||
configure_nvm_env
|
||
|
||
# 激活 nvm
|
||
export NVM_DIR="$HOME/.nvm"
|
||
if [ -s "$NVM_DIR/nvm.sh" ]; then
|
||
. "$NVM_DIR/nvm.sh"
|
||
fi
|
||
if [ -s "$NVM_DIR/bash_completion" ]; then
|
||
. "$NVM_DIR/bash_completion"
|
||
fi
|
||
|
||
# 配置 nvm 镜像源（淘宝源）
|
||
info "配置 nvm 淘宝镜像源..."
|
||
echo 'export NVM_NODEJS_ORG_MIRROR="https://npmmirror.com/mirrors/node"' >> "$HOME/.bashrc"
|
||
if command -v zsh &> /dev/null; then
|
||
echo 'export NVM_NODEJS_ORG_MIRROR="https://npmmirror.com/mirrors/node"' >> "$HOME/.zshrc"
|
||
fi
|
||
export NVM_NODEJS_ORG_MIRROR="https://npmmirror.com/mirrors/node"
|
||
|
||
# 安装 Node LTS 稳定版
|
||
info "安装 Node LTS 稳定版..."
|
||
# 直接安装 LTS 稳定版，无需手动解析版本号
|
||
nvm install --lts || error "Node LTS 版本安装失败"
|
||
# 设置默认版本为 LTS
|
||
nvm alias default lts/* || error "设置 Node 默认版本失败"
|
||
|
||
# 配置 npm 淘宝源
|
||
info "配置 npm 淘宝源..."
|
||
npm config set registry https://registry.npmmirror.com/ || error "配置 npm 源失败"
|
||
# 持久化 npm 配置（全局生效）
|
||
npm config set registry https://registry.npmmirror.com/ --global
|
||
|
||
info "NVM 稳定版安装并配置完成"
|
||
# # 安装 Node LTS 稳定版（跳过 pre-release）
|
||
# info "安装 Node LTS 稳定版..."
|
||
# # 获取 Node LTS 稳定版版本号（跳过 pre）
|
||
# NODE_LTS_VERSION=$(nvm ls-remote --lts | grep -v "-rc" | grep -v "-beta" | tail -1 | awk '{print $1}')
|
||
# if [ -z "$NODE_LTS_VERSION" ]; then
|
||
# # 备用：直接安装 lts
|
||
# nvm install --lts || error "Node LTS 版本安装失败"
|
||
# else
|
||
# nvm install "$NODE_LTS_VERSION" || error "Node LTS 版本安装失败"
|
||
# fi
|
||
# nvm alias default lts/* || error "设置 Node 默认版本失败"
|
||
|
||
# # 配置 npm 淘宝源
|
||
# info "配置 npm 淘宝源..."
|
||
# npm config set registry https://registry.npmmirror.com/ || error "配置 npm 源失败"
|
||
# # 持久化 npm 配置
|
||
# npm config set registry https://registry.npmmirror.com/ --global
|
||
|
||
# info "NVM 稳定版安装并配置完成"
|
||
}
|
||
|
||
# 配置 nvm 环境变量
|
||
configure_nvm_env() {
|
||
info "配置 nvm 环境变量..."
|
||
# 检查 .bashrc 中是否已有 nvm 配置
|
||
if ! grep -q "NVM_DIR" "$HOME/.bashrc"; then
|
||
cat >> "$HOME/.bashrc" << EOF
|
||
|
||
# >>> nvm initialize >>>
|
||
export NVM_DIR="$HOME/.nvm"
|
||
[ -s "\$NVM_DIR/nvm.sh" ] && \. "\$NVM_DIR/nvm.sh"
|
||
[ -s "\$NVM_DIR/bash_completion" ] && \. "\$NVM_DIR/bash_completion"
|
||
# <<< nvm initialize <<<
|
||
EOF
|
||
fi
|
||
|
||
# 配置 zsh（如果存在）
|
||
if command -v zsh &> /dev/null && ! grep -q "NVM_DIR" "$HOME/.zshrc"; then
|
||
cat >> "$HOME/.zshrc" << EOF
|
||
|
||
# >>> nvm initialize >>>
|
||
export NVM_DIR="$HOME/.nvm"
|
||
[ -s "\$NVM_DIR/nvm.sh" ] && \. "\$NVM_DIR/nvm.sh"
|
||
[ -s "\$NVM_DIR/bash_completion" ] && \. "\$NVM_DIR/bash_completion"
|
||
# <<< nvm initialize <<<
|
||
EOF
|
||
fi
|
||
|
||
# 立即生效环境变量
|
||
export NVM_DIR="$HOME/.nvm"
|
||
[ -s "$NVM_DIR/nvm.sh" ] && . "$NVM_DIR/nvm.sh"
|
||
info "nvm 环境变量配置完成"
|
||
}
|
||
|
||
# 验证安装结果
|
||
verify_installation() {
|
||
info "开始验证安装结果..."
|
||
|
||
# 临时关闭未定义变量检查，避免 PS1 报错
|
||
set +u
|
||
# 重新加载环境变量
|
||
source "$HOME/.bashrc"
|
||
# 恢复严格模式
|
||
set -u
|
||
|
||
source "$CONDA_PATH/etc/profile.d/conda.sh"
|
||
|
||
# 验证 conda
|
||
if command -v conda &> /dev/null; then
|
||
info "✅ Mambaconda 安装成功：$(conda --version)"
|
||
info "✅ Python 版本：$(python --version)"
|
||
else
|
||
warn "❌ Mambaconda 验证失败，请手动执行 source $CONDA_PATH/etc/profile.d/conda.sh"
|
||
fi
|
||
|
||
# 验证 nvm/node
|
||
if command -v nvm &> /dev/null; then
|
||
info "✅ NVM 安装成功：$(nvm --version)"
|
||
info "✅ Node 版本：$(node --version)"
|
||
info "✅ NPM 源：$(npm config get registry)"
|
||
else
|
||
warn "❌ NVM 验证失败，请手动执行 source $HOME/.nvm/nvm.sh"
|
||
fi
|
||
}
|
||
|
||
# 清理和提示
|
||
finalize() {
|
||
info "========== 安装配置全部完成 =========="
|
||
info "已安装的稳定版："
|
||
info " - Mambaconda: $LATEST_CONDA_VERSION (Python 3.10)"
|
||
info " - NVM: $LATEST_NVM_VERSION (Node LTS 稳定版)"
|
||
info "请执行以下命令使配置立即生效："
|
||
|
||
if [[ "$OS_TYPE" == "linux" ]]; then
|
||
echo "source $HOME/.bashrc"
|
||
elif [[ "$OS_TYPE" == "macos" ]]; then
|
||
echo "source $HOME/.zshrc"
|
||
fi
|
||
|
||
info "验证安装的命令："
|
||
echo "conda --version && python --version"
|
||
echo "nvm --version && node --version && npm --version"
|
||
echo "npm config get registry && conda config --show-sources"
|
||
}
|
||
|
||
# 主执行流程
|
||
main() {
|
||
info "========== 开始自动安装稳定版 mambaconda 和 nvm =========="
|
||
|
||
# 1. 检测系统信息
|
||
detect_system
|
||
|
||
# 2. 检查用户权限
|
||
check_root
|
||
|
||
# 3. 安装基础依赖
|
||
install_dependencies
|
||
|
||
# 4. 获取最新稳定版版本号（跳过 pre-release）
|
||
get_stable_versions
|
||
|
||
# 5. 安装 mambaconda 稳定版
|
||
install_mambaconda
|
||
|
||
# 6. 安装 nvm 稳定版
|
||
install_nvm
|
||
|
||
# 7. 验证安装结果
|
||
verify_installation
|
||
|
||
# 8. 最终提示
|
||
finalize
|
||
}
|
||
|
||
# 执行主函数
|
||
main |