[Python] Can olmocr Run on Two 12 GB Titan X GPUs?

Discussão em 'Python' iniciado por Stack, Abril 9, 2025.

  1. Stack

    Stack Membro Participativo

    I’m trying to run olmocr (https://github.com/allenai/olmocr) locally, which requires a GPU with 20 GB RAM. I have two Titan X GPUs (12 GB each). When I run it, I get:

    ERROR:eek:lmocr.check:Torch was not able to find a GPU with at least 20 GB of RAM.


    Can I combine their memory or use multi-GPU support to make this work? The code is editable (installed with pip install -e .[gpu]).

    Continue reading...
    Stack, Abril 9, 2025
    #1

Compartilhe esta Página