这个只是警告，没关系
一般是前面报错了。
W0317 23:16:47.236212 4320 device_context.cc:237] Please NOTE: device: 0, CUDA Capability: 61, Driver API Version: 10.2, Runtime API Version: 10.0
W0317 23:16:47.242241 4320 device_context.cc:245] device: 0, cuDNN Version: 7.5.