Skip to content

Conversation

@jobs-git
Copy link

…semantic naming

Fixes #244

Installation of timefs results to installing another set of cuda-toolkit even if the conda environment is compatible with the requirements as indicated in https://github.com/google-research/timesfm/blob/master/pyproject.toml in more particular Jax cuda-12 >= 0.4.26.

This arises because timesfm is specifically asking for jax[cuda12] but conda semantic is instead jax+jaxlib-cuda-12 which pip installer does not understand. The cuda is attached to jaxlib in conda.

Removing cuda12 in jax semantic requirement allows the installation to proceed without messing up the existing ja+cuda-12 installations.

…semantic naming

Fixes google-research#244

Installation of timefs results to installing another set of cuda-toolkit even if the conda environment is compatible with the requirements as indicated in https://github.com/google-research/timesfm/blob/master/pyproject.toml in more particular `Jax cuda-12 >= 0.4.26`.

This arises because `timesfm` is specifically asking for jax[cuda12] but conda semantic is instead jax+jaxlib-cuda-12 which pip installer does not understand. The cuda is attached to jaxlib in conda.

Removing cuda12 in jax semantic requirement allows the installation to proceed without messing up the existing ja+cuda-12 installations.
@google-cla
Copy link

google-cla bot commented Feb 19, 2025

Thanks for your pull request! It looks like this may be your first contribution to a Google open source project. Before we can look at your pull request, you'll need to sign a Contributor License Agreement (CLA).

View this failed invocation of the CLA check for more information.

For the most up to date status, view the checks section at the bottom of the pull request.

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

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

Jax cuda package causes incompatibility with conda environment

1 participant