KidsWithTokens / MedSegDiff

Medical Image Segmentation with Diffusion Model
MIT License
979 stars 147 forks source link

如果我只有一块显卡的话也可以直接用吗,我看代码都是分布式的代码,还是可以把分布式改一下? #108

Closed yuan5828225 closed 1 year ago

Issues-translate-bot commented 1 year ago

Bot detected the issue body's language is not English, translate it automatically. 👯👭🏻🧑‍🤝‍🧑👫🧑🏿‍🤝‍🧑🏻👩🏾‍🤝‍👨🏿👬🏿


Title: If I only have one graphics card, can I use it directly? I think the codes are all distributed codes, or can I change the distributed ones?

WuJunde commented 1 year ago

no need to modify, run in default for single gpu, set -multi_gpu parameter for multiple gpu

yuan5828225 commented 1 year ago

no need to modify, run in default for single gpu, set -multi_gpu parameter for multiple gpu Thank you for your answer