1.本机环境
运行环境:tensorfow1.14, CUDA10.0, Cudnn7.6.5.32,
GPU:1660Ti, 显存6G,
2.运行报错
"Failed to get convolution algorithm. This is probably because cuDNN failed to initialize, so try looking to see if a warning log message was printed above."
3.解决方法
开头添加代码如下:
对GPU内存合理分配,不至于沾满。
from tensorflow import ConfigProto
from tensorflow import InteractiveSession
config = Con