今日运行tensorflow,报出以下错误:
cudaGetDevice() failed. Status: cudaGetErrorString symbol not found.
因为项目需求,之前的tensorflow版本发生了变化。
本机cuda version 是 10.1,之前的项目所需版本不支持cuda10.1.
执行下面两个命令,重装tensorflow:
pip install tensorflow==2.1.0
pip install tensorflow-gpu==2.1.0
运行程序:
OK