abetlen / abetlen/llama-cpp-python
Specify GPU Selection (e.g., CUDA:0, CUDA:1)
未關閉
- 主要語言
- Python
- 星號
- 10.6k
- 分支
- 1.4k
- PR 合併指標
- PR 指標待擷取
描述
Hi,
Is there a way to specify which GPU to use for inference, such as restricting it to only `cuda:0` or `cuda:1` in the code? Or are there any workarounds for achieving this?
Thanks in advance.
貢獻指南
評估
這個 Issue 還沒有評估資料。