Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Add a requirements file for multi-backend cuda #472

Merged
merged 1 commit into from
Jul 16, 2023

Commits on Jul 16, 2023

  1. Add a requirements file for multi-backend cuda

    Not totally sure if we should merge this now, or wait for tf 2.14, but
    figured I could put it up anyway so people could use it. With
    tensorflow/tensorflow#59825
    tf-nightly can be installed using cuda pip packages. Which means we
    can write a recipe for cross framework GPU support.
    
    To install a local development version...
    ```shell
    pip install -r requirements-cuda.txt
    python pip_build.py --install
    ```
    
    To install the official pip version...
    ```shell
    pip install -r requirements-cuda.txt
    pip install keras-core --no-deps
    ```
    
    Note that `--no-deps` is required to avoid pulling in `tensorflow` and
    `tf-nightly` at the same time.
    
    This should work in a clean python env, as long nvidia drivers are
    >=520.61.05. No conda or cuda shenanigans required!
    mattdangerw committed Jul 16, 2023
    Configuration menu
    Copy the full SHA
    bfc166c View commit details
    Browse the repository at this point in the history