ollama/ollama-python
1
0
Fork 0
mirror of https://github.com/ollama/ollama-python.git synced 2026-07-21 00:55:21 -04:00
Code Issues 101 Packages Projects Releases 10 Wiki Activity

GPU Selection & Device Specification #87

New Issue
Open
opened 2026-02-15 16:27:55 -05:00 by yindo · 1 comment
yindo commented 2026-02-15 16:27:55 -05:00
Owner
Copy Link
Copy Source

Originally created by @nycameraguy on GitHub (May 20, 2024).

I would like to suggest a feature that would allow specifing which GPU or GPUs to run on directly within the Ollama Python library.

This feature is crucial in shared server environments across multiple GPUs and multiple users, as it allows each Jupyter notebook to run on the corresponding GPU without conflicts. Currently, specifying GPU usage in Ollama is somewhat complex. A streamlined method to assign tasks to specific GPUs directly inside the Python program would prevent conflicts and optimize workflow. Implementing this feature would significantly improve usability and align Ollama with other machine-learning frameworks.

Thank you for considering this suggestion. I would be happy to discuss further details if needed.

Originally created by @nycameraguy on GitHub (May 20, 2024). I would like to suggest a feature that would allow specifing which GPU or GPUs to run on directly within the Ollama Python library. This feature is crucial in shared server environments across multiple GPUs and multiple users, as it allows each Jupyter notebook to run on the corresponding GPU without conflicts. Currently, specifying GPU usage in Ollama is somewhat complex. A streamlined method to assign tasks to specific GPUs directly inside the Python program would prevent conflicts and optimize workflow. Implementing this feature would significantly improve usability and align Ollama with other machine-learning frameworks. Thank you for considering this suggestion. I would be happy to discuss further details if needed.
yindo commented 2026-02-15 16:27:56 -05:00
Author
Owner
Copy Link
Copy Source

@gileneusz commented on GitHub (May 25, 2024):

would be great if this could be implemented asap

@gileneusz commented on GitHub (May 25, 2024): would be great if this could be implemented asap
Sign in to join this conversation.
No Branch/Tag Specified
Branches Tags
Labels
Clear labels
bug
enhancement
pull-request
Mirrored from GitHub Pull Request
 thinking
No Label
Milestone
No items
No Milestone
Projects
Clear projects
No project
Assignees
Clear assignees
yindo
No Assignees
1 Participants
Notifications
Due Date
No due date set.
Dependencies

No dependencies set.

Reference: ollama/ollama-python#87
Delete Branch "%!s()"

Deleting a branch is permanent. Although the deleted branch may continue to exist for a short time before it actually gets removed, it CANNOT be undone in most cases. Continue?