错误一: Check failed: error == cudaSuccess(30 vs. 0)
1. 方法一:使用sudo python 运行即可
2. 方法一:重装显卡驱动,如图在软件中心切换显卡驱动版本完成后重启
错误二:Check failed: status == CUDNN_STATUS_SUCCESS (1 vs. 0) CUDNN_STATUS_NOT_INIT
解决:cudnn版本错误,重新选择正确的版本安装https://developer.nvidia.com/cudnn