docker-immich icon indicating copy to clipboard operation
docker-immich copied to clipboard

FEATURE - support gpu acceleration for machine-learning

Open martabal opened this issue 2 years ago • 5 comments

Immich v1.94.0 will introduce the ability to support GPU acceleration for machine-learning.

Nvidia - CUDA

Cuda acceleration shouldn't be too difficult to support, we should be able to reuse the custom scripts we created a few months ago.

Intel - OpenVINO

Openvino acceleration is not possible at the moment because it requires python 3.10. And with the current base image (ubuntu 23.10), python 3.10 is only available if you compile it. The best solution would be to wait until Ubuntu 24.04 LTS is released and use the Deadsnakes custom PPA to install python 3.10 or onnxruntime-openvino 1.17.

To do :

  • [x] Support cuda
  • [ ] Support openvino

martabal avatar Jan 21 '24 23:01 martabal

OpenVINO might support python 3.11 with the next onnxruntime-openvino release (1.17.0)

martabal avatar Jan 23 '24 13:01 martabal

I tried openvino with python 3.10, but runned into issues with onnxruntime.

Edit: to be more precise : https://github.com/immich-app/immich/discussions/6869

martabal avatar Feb 08 '24 18:02 martabal

Update:

  • Immich v1.98.0 will no longer need the nightly version of onnxruntime
  • I have a working version for Openvino but it must compile python 3.10 which is not a viable option for a docker mod (it is not realistic to wait 5 minutes for a docker to start)

martabal avatar Mar 07 '24 18:03 martabal

Update:

  • Immich v1.98.0 will no longer need the nightly version of onnxruntime
  • I have a working version for Openvino but it must compile python 3.10 which is not a viable option for a docker mod (it is not realistic to wait 5 minutes for a docker to start)

compile then package? or a deb?

hydazz avatar Mar 13 '24 10:03 hydazz

compile then package? or a deb?

Not sure to correctly understand what you mean. Having a good python deb is hard, I would prefer to wait ubuntu 24.04 LTS.

martabal avatar Mar 13 '24 10:03 martabal

Hi there!

Any updates for this, for Intel iGPUs? Thanks :)

myanesp avatar Jun 06 '24 10:06 myanesp

Hello!

Currently, OpenVINO is broken https://github.com/immich-app/immich/issues/8226 (at least on my machines), so I can't really test it.

Once the dependency and the base image are updated, I'll implement it.

martabal avatar Jun 06 '24 11:06 martabal

Done with #392

martabal avatar Jul 03 '24 15:07 martabal