abetlen / abetlen/llama-cpp-python
Specify GPU Selection (e.g., CUDA:0, CUDA:1)
Đang mở
- Ngôn ngữ chính
- Python
- Star
- 10.6k
- Fork
- 1.4k
- Chỉ số merge pull request
- Chỉ số pull request đang chờ
Mô tả
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.
Hướng dẫn đóng góp
Đánh giá
Issue này chưa được đánh giá.