mpiexec runs on colab pro?

[aditipanda]

I changed compute_dire.sh to have only 2 gpus instead of four (as written by default in the sh file): export CUDA_VISIBLE_DEVICES=0,1 but still my run is slow and not computing dire images fast enough. I'm connected to a T4 High RAM GPU on Google Colab Pro.

Has anyone faced this problem? Pls help.