kuixu / alphafold

Install alphafold on the local machine, get out of docker.
Apache License 2.0
97 stars 34 forks source link

Only 1 GPU is used #3

Open rui--zhang opened 3 years ago

rui--zhang commented 3 years ago

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!