Slurm集群启动来自自己节点的Conda环境

学校提供了硬件牛掰的集群,然而用了slurm管理。
直接conda activate显示找不到对应环境,在网上转了一圈没找到合适的办法。
后来一拍脑门,想到节点找不到环境的主要原因是找不到自己安装的conda,那就把bash.rc的启动代码搬进slurmrun.sh脚本,问题解决!

以下是一个slurmrun.sh脚本样例,供诸君避坑的q(≧▽≦q)

#!/bin/sh
#SBATCH --partition=standard
#SBATCH --gpus=1
#SBATCH --time=30
#SBATCH --job-name=myjob

# 以下部分的路径请全部换成你的conda安装的路径
__conda_setup="$('/home/.../miniconda3/bin/conda' 'shell.bash' 'hook' 2> /dev/null)"
if [ $? -eq 0 ]; then
    eval "$__conda_setup"
else
    if [ -f "/home/.../miniconda3/etc/profile.d/conda.sh" ]; then
        . "/home/.../miniconda3/etc/profile.d/conda.sh"
    else
        export PATH="/home/.../miniconda3/bin:$PATH"
    fi
fi
unset __conda_setup

# 启动你的环境,记得替换成你的环境的名称
conda activate your-env-name

# 启动你的python程序脚本
python test.py
  • 4
    点赞
  • 1
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值