错误如下:
错误原因:利用CPU版本的tensorflow时调用了cuda,可能是因为服务器行装了cuda及cudnn
解决方案:
在脚本的开头添加如下代码:
import os
os.environ["CUDA_VISIBLE_DEVICES"]="-1"
错误如下:
错误原因:利用CPU版本的tensorflow时调用了cuda,可能是因为服务器行装了cuda及cudnn
解决方案:
在脚本的开头添加如下代码:
import os
os.environ["CUDA_VISIBLE_DEVICES"]="-1"