search

pnlbwh / CNN-Diffusion-MRIBrain-Segmentation

CNN based brain masking
Other
14 stars 10 forks source link

Use CUDA_VISIBLE_DEVICES #37

Closed tashrifbillah closed 11 months ago

tashrifbillah commented 1 year ago

Current setup defaults to only GPU 0. This setup seems to run into memory errors during concurrent mask creation. So we route different cases to different GPUs.

tashrifbillah commented 11 months ago

Superseded by PR #38