tensorflow2.0-GPU 安装之后遇到的问题

配置环境

windows 10
python 3.7.5
tensorflow2.0-GPU
cuda 10.0
cudnn 7.6.5
CPU i5 9400f
GPU RTX2060

1.GpuLaunchKernel

GpuLaunchKernel(FillPhiloxRandomKernelLaunch, num_blocks, block_size, 0, d.stream(), gen, data, size, dist) status: Internal: invalid device function.

解决办法
我在配置环境时，最开始安装的是cuda10.1,卸载cuda10.1之后，安装cuda10.0已经对应的cudnn就解决这个问题了。

2.cuDNN failed to initialize

UnknownError: 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. [Op:Conv2D]
解决办法
这个问题可能是显卡内存爆炸的问题，限制一下显卡内存就行。在程序中加入以下代码。

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=5000)]  # 数值根据显卡内存设定