Skip to content

Are there some GPU ready configurations and related documentation? #101

Open
Open
Are there some GPU ready configurations and related documentation?#101

Description

@mhmaqbool
mhmaqbool
opened on Sep 4, 2024

Hi,

I want to run the models on GPU. I was wondering if there is some GPU ready configurations or any related documentation provided for that matter?

Thanks.

Activity

  1. mhmaqbool
    changed the title [-]Are there some GPU ready configuration and related documentation?[/-] [+]Are there some GPU ready configurations and related documentation?[/+] on Sep 4, 2024

  2. entrpn commented on Sep 5, 2024

    @entrpn
    entrpn
    on Sep 5, 2024
    Collaborator

    I haven't done extensive testing with GPUs yet. You can try running the same installation instructions, but installing jax[cuda] pip install -U "jax[cuda12]" instead of jax[tpu].

  3. entrpn commented on Sep 12, 2024

    @entrpn
    entrpn
    on Sep 12, 2024
    Collaborator

    @mhmaqbool did this help?

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Metadata

Metadata

Assignees

No one assigned

    Labels

    No labels
    No labels

    Type

    No type

    Projects

    No projects

      Milestone

      No milestone

      Relationships

      None yet

      Development

      No branches or pull requests

      Issue actions