search

pnlbwh / CNN-Diffusion-MRIBrain-Segmentation

CNN based brain masking
Other
14 stars 10 forks source link

Algorithm for using preferred GPU device #38

Closed tashrifbillah closed 11 months ago

tashrifbillah commented 11 months ago

Three cases are covered:

  1. User does not specify:

unset CUDA_VISIBLE_DEVICES

(output is 0, default)

  1. User specifies:

export CUDA_VISIBLE_DEVICES=1

(output is 1)

  1. User specifies an invalid GPU:

export CUDA_VISIBLE_DEVICES=7

(output is 0, default)

@RyanZurrin, can you do the tests using:

/rfanfs/pnl-zorro/software/pnlpipe3/CNN-Diffusion-MRIBrain-Segmentation/pipeline/dwi_masking.py -h

tashrifbillah commented 11 months ago

Ryan, can you review this PR?

RyanZurrin commented 11 months ago

All Tests had the correct GPU device output.

tashrifbillah commented 11 months ago

We shall merge it after processing 4 EDCRP cases parallelly.