在keras中使用gpu加速训练模型;安装cuda;cudnn;cudnn_cnn_infer64_8.dll 不在path中;device_lib.list_local_devices无gpu;挂掉
首先打开任务管理器,查看自己电脑是否存在gpu注意,以前安装的tensorflow默认是cpu版的,所以我们下载这个是gpu版的,注意版本得和tensorflow一致(网上说要在tensorflow安装之后安装tensorflow-gpu)首先在命令行输入 NVIDIA-smi,查看自己的gpu,驱动版本,CUDA版本打开网址 https://developer.nvidia.com/cuda-toolkit-archive按照以下表格安装对应的CUDA版本(我的11.5是已经下载好的),下载exe
原创
2022-06-07 19:05:39 ·
3520 阅读 ·
5 评论