Page 1 of 1

How to make it run on GPU

Posted: Wed Apr 29, 2020 7:23 am
by Kevin_Z

My CPU is AMD and GPU is NVIDIA. The extraction is all on CPU and quite slow. The resolution is 256x256


Re: How to make it run on GPU

Posted: Wed Apr 29, 2020 9:01 am
by torzdf

Extraction uses both CPU and GPU


Re: How to make it run on GPU

Posted: Wed Apr 29, 2020 10:16 am
by Kevin_Z
torzdf wrote: Wed Apr 29, 2020 9:01 am

Extraction uses both CPU and GPU

My CUDA doesn't not match Tensorflow, I fixed it. THX!