Question about the multi-gpu running: 'mpirun -np ...'#77
Closed
Thanks for your great work! When I run the code with multi-gpu, I met a problem: When using a single GPU, the code runs normally. However, when running with multiple GPUs using the command 'mpirun -np 2 python finetune_sdm_yaml.py ...', it is observed that all processes are on GPU: 0.  I found the problem I met is the same as #48. However, I tried the method mentioned in this issue, but it didn't solve the problem. ps: I note that the code run into the following part in utils/dist.py: 
AI Analysis
This issue appears to be discussing a feature request or bug report related to the repository. Based on the content, it seems to be resolved. The issue was opened by CHNxindong and has received 7 comments.
Add a comment
Comment form would go here