Differentiate instruction for CPU vs GPU support - Githubissues

pnlbwh / CNN-Diffusion-MRIBrain-Segmentation

CNN based brain masking

Other

14 stars 10 forks source link

Differentiate instruction for CPU vs GPU support #9

Closed tashrifbillah closed 4 years ago

tashrifbillah commented 4 years ago

Should it be:

For CPU:

pip install tensorflow==1.12.0

For GPU:

pip install tensorflow-gpu==1.12.0
conda install cudatoolkit=9.0
conda install cudnn=7.0.5

For both:

conda install -c pnlbwh ants
pip install keras==2.2.4
pip install nibabel
pip install gputil

senthilcaesar commented 4 years ago

Got it . Thank you Tashrif

tashrifbillah commented 4 years ago

Fixed by https://github.com/pnlbwh/CNN-Diffusion-MRIBrain-Segmentation/commit/fdad921373b6a95360fa975394845e55d0b63bd9