一、问题
centos8升级显卡运行tensorflow报错 CUDA Error: no kernel image is available for execution on device
(1)原有版本为:
tf:2.3
cuda:10.1
cudnn:7.5
(2)升级后版本
tf:2.4
cuda:11.1
cudnn:8.04
二、原因分析
cuda与tensorflow的版本不匹配
三、解决方法
升级tensorflow至2.4
pip install -U -I setuptools
pip install -U tensorflow