How do I run test.py with multiple gpus

chaoyi-wu / RadFM

The official code for "Towards Generalist Foundation Model for Radiology by Leveraging Web-scale 2D&3D Medical Data".

315 stars 32 forks source link

Open dontyougetthere opened 3 months ago

dontyougetthere commented 3 months ago

The memory of my single gpu is not enough to load the whole model. Could you give me a plan to store it on multiple Gpus？