Skip to content
This repository was archived by the owner on May 28, 2024. It is now read-only.
This repository was archived by the owner on May 28, 2024. It is now read-only.

TF 2.5+ GPU build no longer works because CUDA 11 configure is not supported #105

Open

Description

@trevor-m

With TensorFlow 2.5, there have been some code changes which require CUDA 11. You can read more about the issues when trying to build TF 2.5 against CUDA 10 here:
tensorflow/tensorflow#49983
tensorflow/tensorflow#48468

The build rules to configure CUDA in this custom-op repo (https://github.com/tensorflow/custom-op/tree/master/gpu) have not been updated in a long time. Because of that, it is not possible to compile against newer CUDA versions so we can no longer build our custom gpu plugin against newer TF releases.

Questions

  • Will this repository be updated to support newer versions of CUDA? Otherwise, it is impossible to build custom gpu ops with newer TF versions (2.5+).
  • There is an alternative way to build custom ops through using the main TensorFlow repository by placing the custom op code in this folder https://github.com/tensorflow/tensorflow/tree/master/tensorflow/core/user_ops which will reuse the configuration code from the main tensorflow repository. Is this the recommended approach moving forward and will this repository be deprecated? Is this approach equivalent in terms of what kind of plugins can be built?

Thanks!

Activity

  1. seanpmorgan commented on Jul 15, 2021

    @seanpmorgan
    Member

    Hey @trevor-m while this repo appears to be stale, there aren't too many modification that need to be made to support TF2.5+. You can check the changes that addons made:
    https://github.com/tensorflow/addons/pull/2441/files
    https://github.com/tensorflow/addons/pull/2218/files

    https://github.com/tensorflow/addons/tree/master/build_deps

  2. seanpmorgan commented on Jul 15, 2021

    @seanpmorgan
    Member

    Alternatively it does look like you can use that user_ops directory if you're compiling TF from source and not a plugin library

  3. trevor-m commented on Jul 15, 2021

    @trevor-m
    Author

    Hi @seanpmorgan, thanks for the response and for sharing the TF/addons repostiory. Looks like there is also one more PR for cuda 11.
    tensorflow/addons#1950
    Will this repository also be updated with those changes?

    Alternatively it does look like you can use that user_ops directory if you're compiling TF from source and not a plugin library

    I see, so it is not possible to build a plugin using the user_ops directory?

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