在云平台跑模型一个月花了我七八百,一咬牙自己配了一台主机。问题来了,使用pycharm占内存不说,将控制台的输出重定向到文件中也是挺麻烦的事情。还是怀念linux平台命令行的简单粗暴。于是我琢磨出直接用windows的控制台无限去接近在linux上的习惯。
抛弃那些冗余的重定向代码,在miconda中激活自己的环境,并将工作目录cd 到你的项目目录。以我的项目为例:
(base) C:\Users\PC>conda activate pc_torch
(pc_torch) C:\Users\PC>
(pc_torch) C:\Users\PC>cd /d D:\binocular\IGEV\IGEV-main - fix\IGEV-Stereo
(pc_torch) D:\binocular\IGEV\IGEV-main - fix\IGEV-Stereo>
别忘了/d没有的话不能跨盘切换目录。然后直接运行train.py并进行重定向
(pc_torch) D:\binocular\IGEV\IGEV-main - fix\IGEV-Stereo>python train_stereo.py > train.log 2>&1
下面这个是核心代码:
python xxx.py > train.log 2>&1
用的和linux一样的重定向命令,没想到这个windows也能用,最后程序就愉快的跑起来了,重定向就捎带完成了,个人感觉比在pycharm项目里去写Logger 类和用sys去重定向方便。
用pycharm跑:
用系统控制台跑:
看这内存占用率pycharm比cmd多占用一倍不止。如果您感觉有用的话请给个赞吧,双非谁水硕小菜菜,有啥班门弄斧的地方,大佬们轻喷。