addons icon indicating copy to clipboard operation
addons copied to clipboard

Published 20 hours ago • verified publisher icon tensorflow

CUDA 11.7 support

Open ghylander opened this issue 3 years ago • 1 comments

  • OS Platform and Distribution (e.g., Linux Ubuntu 16.04): Ubuntu 20.04
  • TensorFlow version and how it was installed (source or binary): 2.9.1-binary
  • TensorFlow-Addons version and how it was installed (source or binary): 0.17.1-binary
  • Python version: 3.8
  • Is GPU used? (yes/no): yes

Currantly, simply importing tfa in a script on a system with CUDA 11.7 will result at some point in a seg fault. Even without calling any funcion or object. What would need to be done to add support for CUDA 11.7?

ghylander avatar Sep 19 '22 07:09 ghylander

I think that we have only tested it with this compatibility matrix:

https://github.com/tensorflow/addons#c-custom-op-compatibility-matrix

Can you try with one of these?

bhack avatar Sep 27 '22 11:09 bhack

Per compatibility matrix, you'll need to compile from source in order to run GPU custom op with CUDA 11.7

seanpmorgan avatar Mar 01 '23 04:03 seanpmorgan