How to use multi-gpu? - Githubissues

taigw / brats17

Brain tumor segmentation for MICCAI 2017 BraTS challenge

BSD 3-Clause "New" or "Revised" License

321 stars 130 forks source link

How to use multi-gpu? #26

Open joe1chief opened 5 years ago

joe1chief commented 5 years ago

This program seems to only run on one GPU . Can anyone tell me how to run with multiple GPUs? I have assigned GPU 2 and 3 with instruction "export CUDA_VISIBLE_DEVICES=2,3 ".

taigw commented 5 years ago

Hi, the repository has only been tested with a single gpu.