Bashrc 中的自定义函数

由于博主经常需要在新的终端里面设置环境变量，比较麻烦，因此设置一下常见的函数，后期会不断扩充。

# 设置代理
p1() {
    local port="${1:-7890}"
    export http_proxy="http://127.0.0.1:${port}"
    export https_proxy="http://127.0.0.1:${port}"
    echo "代理已开启 (127.0.0.1:${port})"
}

# 关闭代理
p0() {
    unset http_proxy
    unset https_proxy
    echo "代理已关闭"
}

p2() {
    curl https://www.youtube.com
}

# 设置 CUDA 可见设备
cuda_on() {
    if [ -z "$1" ]; then
        echo "用法: cuda_on <设备号>"
        echo "示例: cuda_on 0    # 使用 GPU 0"
        echo "      cuda_on 0,2,4  # 使用 GPU 0,2,4"
        return 1
    fi
    export CUDA_VISIBLE_DEVICES="$1"
    echo "CUDA_VISIBLE_DEVICES 已设置为: $1"
}

# 恢复所有 GPU 可见
cuda_off() {
    unset CUDA_VISIBLE_DEVICES
    echo "已恢复所有 GPU 可见"
}

# 查看当前 CUDA 设备设置
cuda_show() {
    if [ -z "$CUDA_VISIBLE_DEVICES" ]; then
        echo "当前未设置 CUDA_VISIBLE_DEVICES（所有 GPU 可见）"
    else
        echo "当前 CUDA_VISIBLE_DEVICES = $CUDA_VISIBLE_DEVICES"
    fi
}

# 环境相关
# 保存当前目录
csave() {
  pwd > ~/.last_work_dir
  echo "已保存目录：$(pwd)"
}

# 进入保存的目录，可选激活 conda 环境
# 用法：
#   cgo
#   cgo myenv
cgo() {
  if [ ! -f ~/.last_work_dir ]; then
    echo "还没有保存过目录，请先执行 csave"
    return 1
  fi

  cd "$(cat ~/.last_work_dir)" || return
  
  conda activate dl

}