JupyterNotebook运行Python代码实现传参方式.docx
第
JupyterNotebook运行Python代码实现传参方式
目录JupyterNotebook运行Python代码实现传参例如,使用argv[1]时出现例如,在如下命令行中jupyternotebook参数化运行pythonUpdatesNotesCodesinpyfileinnotebook总结
JupyterNotebook运行Python代码实现传参
在JupyterNotebook中,运行Python源代码非常方便,但是如何模拟命令行方式运行时的输入参数呢?
如果直接使用sys.argv会出现错误。
例如,使用argv[1]时出现
导致上述错误的原因为:在JupyterNotebook中运行Python代码时,argv中有三个默认参数,分别是:
argv[0]:‘路径\test.py(源代码文件名)
argv[1]:‘-f
argv[2]:路径\kernel-…json
而在命令行方式下运行Python代码时,argv[0]与上述argv[0]相同,但argv[1]为命令行中紧随在test.py之后的字符串
例如,在如下命令行中
pythontest.pytest.txt
argv[1]为test.txt
因此,解决方案就是对listargv进行修改,如下面的代码所示:
#需要sys模块来使用argv
fromsysimportargv
#查看当前的argv列表
print(len(argv))
print(argv)
#直接修改argv
argv[1]=test.txt
#也可以添加新的参数
argv.append(test.txt)
这样,在后续代码中,.py和.ipynb源代码一致,无需为在JupyterNotebook中运行而进行修改。
jupyternotebook参数化运行python
Updates
(2025.8.1419:53)吃饭前用这个方法实战了一下,吃完回来一看好像不太行:跑完一组参数之后,到跑下一组参数时好像没有释放之占用的GPU,于是notebook上的结果,后面好几条都报错说cudaoutofmemory。
现在改成:将notebook中的代码写在一个python文件中,然后用命令行运行这个文件,比如:
#autorun.py
importos
#print(os.getcwd())
over=[?#之前手工改参数跑完的参数组合
??[0,1,1],[0,1,2],[0,1,3],
??[0,2,1],
??[1,0,1],
??[1,2,1]
foralphainrange(1,4,1):
??forbetainrange(3):
????forgammainrange(3):
??????if[alpha,beta,gamma]inover:
????????continue
??????os.system(fpythonmain.py--alpha{alpha}--beta{beta}--gamma{gamma})
这里的main.py是训练用的主文件。改在py里用os.system跑,希望跑一组参数之后完会自动释放资源再跑下一组()
Notes
有多组参数组合需要尝试,不想每组参数都人工修改python代码,再在notebook中%run它。
python参数通过的argparse接收,在notebook中写个多重循环遍历参数组合传给python程序自动运行。
记录一个简例。
Codes
test_dir
|-test.py
|-test.ipynb
inpyfile
#test.py
importargparse
parser=argparse.ArgumentParser()
parser.add_argument(--number,type=int,default=0,help=number)
parser.add_argument(--string,type=str,default=abc,help=string)
args=parser.parse_args()
print(number:,args.number,type(args.number))
print(st