Program perfectly runs on CPU but gives value error when tested on GPU

DeepTrial / Retina-VesselNet

A Simple U-net model for Retinal Blood Vessel Segmentation based on tensorflow2

305 stars 76 forks source link

Program perfectly runs on CPU but gives value error when tested on GPU #17

Closed aakash-saboo closed 3 years ago

aakash-saboo commented 5 years ago

Hi! I tried to test this code on GPU but it gives value error. The program runs perfectly fine on CPU but takes a lot of time(as it is expected). I am adding the screenshot of the error I am using tensorflow 1.12 and keras 2.2.4 Also please see this link for a potential guidence.

Also , i request you to mention which version of tensorflow and keras did you use.

screenshot from 2019-01-19 12-11-45

Thank you.

aakash-saboo commented 5 years ago

I installed tensorflow 1.4,keras 2.2.1 and cuda 8.0 and cudnn 6. and it worked fine.