配置环境
windows 10
python 3.7.5
tensorflow2.0-GPU
cuda 10.0
cudnn 7.6.5
CPU i5 9400f
GPU RTX2060
问题描述
Fail to find the dnn implementation
问题解决办法
在程序中加入下列代码
gpus = tf.config.experimental.list_physical_devices(device_type='GPU')
tf.config.experimental.set_memory_growth(gpus[0], True)
tf.config.experimental.set_virtual_device_configuration(
gpus[0],
[tf.config.experimental.VirtualDeviceConfiguration(memory_limit=3000)] # 数值由显卡内存决定
)