如何在screen命令中配置THEANO_FLAGS以正常运行Theano程序?
解决screen中带Theano环境变量的命令执行问题
这事儿我之前也碰到过!问题出在screen对命令的解析方式上——它默认不会像你平时用的交互式shell那样处理环境变量赋值+命令这种组合语法,直接写的话会把整个环境变量字符串当成命令名,自然执行失败。给你几个靠谱的解决办法:
方法1:让bash负责解析整个命令
把环境变量赋值和python命令打包成一个字符串,让bash -c来执行,这样shell就能正确识别环境变量赋值了:
screen -dmSL screen_name bash -c "THEANO_FLAGS='mode=FAST_RUN,device=cuda,floatX=float32,optimizer_including=cudnn' python myfile.py"
这里外层用双引号把整个命令包起来,内层THEANO_FLAGS用单引号防止里面的逗号等字符被shell误解析,完美避开语法问题。
方法2:先用export设置环境变量
如果觉得直接赋值看着别扭,也可以先在screen的子shell里导出环境变量,再执行python脚本:
screen -dmSL screen_name sh -c 'export THEANO_FLAGS="mode=FAST_RUN,device=cuda,floatX=float32,optimizer_including=cudnn"; python myfile.py'
这个逻辑更直观,先把变量设好,再跑脚本,效果和方法1一样。
方法3:全局配置环境变量(适合频繁使用的场景)
如果你大部分Theano脚本都需要这个环境变量,直接把它加到你的shell配置文件里(比如~/.bashrc或者~/.profile):
export THEANO_FLAGS='mode=FAST_RUN,device=cuda,floatX=float32,optimizer_including=cudnn'
保存后重新加载配置(source ~/.bashrc),之后再用screen启动脚本就可以回到你原来的简单命令了:
screen -dmSL screen_name python myfile.py
不过这个方法是全局生效的,要是你有个别脚本需要不同的Theano配置,还是前两种方法更灵活。
小提醒
如果你的环境变量值里有特殊字符(比如空格、引号),一定要注意引号的嵌套规则——外层用双引号时,内层可以用单引号;反过来也行,只要保证整个命令能被shell正确解析就好。
内容的提问来源于stack exchange,提问作者bhomass
相关产品推荐
相关产品推荐

