GPU显存问题:0号卡会叠加进程并占用额外的显存

无论是我自己的代码,还是使用官方的demo测试:
mpirun -np 4 python3.7 -m jittor.test.test_resnet
都会发现显存占用不均匀的问题,0号卡会额外有进程占用并占用一定量的显存,这使得我无法开到更大的batchsize