search

tensorflow / custom-op

Guide for building custom op for TensorFlow
Apache License 2.0
376 stars 115 forks source link

support for CUDA 11.7? #113

Closed ghylander closed 2 years ago

ghylander commented 2 years ago

Currently, the highest supported CUDA version is 11.2. Som eof the classes I use from tfa result in a seg fault when running under linux, with TF 2.9 and python 3.8 and CUDA 11.7 Any plans to update this, or any indications on how to tackle this issue?

ghylander commented 2 years ago

Just realised i posted this in the wrong repo, it was meant for the tensorflow-addons repo