在.slurm文件中激活Anaconda环境

超算中心使用slurm作为集群调度。原始slurm脚本如下:

source activate tensorflow-gpu
python neural_style.py --content content/nju-1.jpg --styles style/18.jpg --output result-v100/result-nju-18.jpg

srun -p gpu -w g0051提交任务后报错:

很明显,第一句激活anaconda环境失败了,提示找不到conda activate可执行程序。

Google了一下,发现了这个:

https://github.com/conda/conda/issues/5071

看了一下,大概心里有数了,直接将slurm脚本改为:

source /home/xieweibo/project/software/anaconda3/bin/activate tensorflow-gpu
python neural_style.py --content content/nju-1.jpg --styles style/18.jpg --output result-v100/result-nju-18.jpg

也就是将conda activate可执行程序作为绝对路径在任务脚本中索引即可。

任务成功提交至调度系统:

 

tz@croplab, hzau

2019-11-23

posted on 2019-11-23 15:04  tuzhuo  阅读(4207)  评论(0编辑  收藏  举报