You signed in with another tab or window. Reload to refresh your session.You signed out in another tab or window. Reload to refresh your session.You switched accounts on another tab or window. Reload to refresh your session.Dismiss alert
{{ message }}
This repository was archived by the owner on May 28, 2024. It is now read-only.
Repository navigation
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
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?
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?
Activity
seanpmorgan commentedon Jul 15, 2021
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
seanpmorgan commentedon Jul 15, 2021
Alternatively it does look like you can use that user_ops directory if you're compiling TF from source and not a plugin library
trevor-m commentedon Jul 15, 2021
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?
I see, so it is not possible to build a plugin using the user_ops directory?