RTX3090 tf2 单机多卡 Check failed: context != nullprt 问题

技术问答
1

rtx3090 tf2.4.1 单机多卡，代码运行报错（同样的代码在rtx2080上正常运行），报错为：
F tensorflow/stream_executor/cuda/cuda_driver.cc:1289] Check failed: context != nullptr success should entail non-null context。
具体是，当运行到model class中的 call(self, input)的时候会报这个错误，即将用掉显卡计算input。
大家知道怎么解决吗？

1 Like
2

TF 2.5 nightly CUDA driver check failed · Issue #46851 · tensorflow/tensorflow · GitHub 参考一下这个，降级TensorFlow和CUDA试试，感觉可能是新版Bug