Only 1 GPU is used

[rui--zhang]

Hi, I could successfully install and run your version of alphafold, however, I notice only 1 GPU is used, even though I have 8x RTX 3090 cards.
When it reaches certain stage of relaxation, all the CPUs are used.

I tried to put "export CUDA_VISIBLE_DEVICES=0,1,2,3,4,5,6,7" in exp/run_local.sh, but still only 1 GPU is used.
Any suggestion is greatly appreciated!