failed call to cuDevicePrimaryCt
2020-07-24 本文已影响0人
我的章鱼小丸子呢
报错信息：
错误
原因：
利用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"
