报错如下:tensorflow.python.framework.errors_impl.InternalError: cudaGetDevice() failed. Status: CUDA driver version is insufficient for CUDA runtime version
原因:CUDA驱动版本不满足CUDA运行版本,要更新一下显卡驱动
方法:
- 找到驱动:
Nvidia官网:https://www.nvidia.cn/Download/index.aspx?lang=cn
输入显卡系列、产品名称、os、cuda版本等等信息
查看方法如下
我的显卡名称是NVIDIA Geforce 920M
搜索后下载最新版本的显卡驱动,然后(精简)安装即可
检查:
选择“运行”,在“运行”里面输入“dxdiag”命令,发现已更新。
重新运行train_anti.py文件,开始训练模型
早上起来时60轮训练完毕,周末good morning!