devika
devika copied to clipboard
fix: Enable NVIDIA GPU support for Ollama in Docker
added NVIDIA GPU support for OLLAMA in docker-compose.yaml
can make the latest docker files that compatible with the current changes.
This code throws error. if nvidia gpu container toolkit is not present ( which will prevent non-gpu from running docker).
I would suggest to handle it gracefully using "runtime" and a environment variable in code, so that it will give flexibity for a non-GPU user also to run & also it won't throw errors if nvidia-gpu-container- toolkit is not present.
So it can handle in better way? Or I just skip this pr currently?