search

mshunshin / SegNetCMR

A Tensorflow implementation of SegNet for cardiac MRI segmentation
MIT License
74 stars 35 forks source link

No OpKernel was registered to support Op 'MaxPoolWithArgmax' #2

Closed DawnMe closed 7 years ago

DawnMe commented 7 years ago

After running python train.py, I'm getting the following error:

No OpKernel was registered to support Op 'MaxPoolWithArgmax' with these attrs. Registered devices: [CPU], Registered kernels:

Anyone has encountered this? Shahar
mshunshin commented 7 years ago

Yep - this isn't available on CPUs - you need a GPU to run this.

See: https://github.com/tensorflow/tensorflow/issues/6035

DawnMe commented 7 years ago

@mshunshin Thanks. Closed the issue.