Tensorflow:UnknownError: Failed to get convolution algorithm. This is probably because cuDNN failed
1.问题
采用Tensorflow 1.14.0报错:
2.解决办法
降低Tensorflow的版本,我的代码会出现更多问题,也尝试过GPU资源分配的方法,均不行
最终解决办法:
将原代码:
os.environ["CUDA_VISIBLE_DEVICES"] = "0"
修改为:
os.environ["CUDA_VISIBLE_DEVICES"] = "gpu:0"
最终困扰了半天的问题解决!啊啊啊啊啊,也不知道为什么!!
2020.7.9更新:
使用os.environ["CUDA_VISIBLE_DEVICES"] = "0"
时,会使用GPU资源
使用os.environ["CUDA_VISIBLE_DEVICES"] = "gpu:0"
时,仅会使用CPU资源