4 ms·
> Does Python have a library that works with the GPU for image processing? tensorflow. It's not just deep learning library and you can implement any tensor ope
by kozikow 9y ago
> Does Python have a library that works with the GPU for image processing?
tensorflow. It's not just deep learning library and you can implement any tensor operations there. 99% image processing is just tensor operations.
- make3 9y agoexcept just loading tensorflow takes half an hour on some machines. pytorch is better in that regards. both are probably too hard to install though, cuda and cudnn are really unpleasant to install, unless you are using conda
- mozumder 9y agoIntriguing option but I'm dealing with the integrated GPU here. It looks like TensorFlow is for NVidia discrete GPUs. PythonOpenCL is probably the best option: https://mathema.tician.de/software/pyopencl/ https://mathema.tician.de/software/pyopencl/ One thing about integrated GPUs for server is that there's no transfer time, since the GPUs share memory with the CPU.
- Marat_Dukhan 9y agoThere is transfer time. You source image is in cacheable CPU memory. Integrated GPUs normally work with uncacheable memory allocated in a special region of system memory. Some GPUs can access cacheable memory too, but it is much slower (because it has to maintain coherency with CPU caches), and requires that you allocate such cacheable memory using special OpenCL driver calls, not your normal malloc. So, in practice, you would do a copy to GPU-optimized buffer (in shared with CPU, but uncacheable memory).