报错信息:
错误
原因:
利用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"