use GPU issue - Githubissues

intel / unet

U-Net Biomedical Image Segmentation

Apache License 2.0

303 stars 124 forks source link

use GPU issue #15

Closed qiuLimoges closed 4 years ago

qiuLimoges commented 4 years ago

Hi, very good job and think you for share. I want to know can we use GPU for tensor calculation? thank you

tonyreina commented 4 years ago

For the GPU you should just be able to use the TensorFlow version compiled for the GPU. I think the only thing that would change is if you wanted to do the Horovod part, but there are GPU implementations using that.