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