Keras深度学习环境配置（WIN10/GPU加速版）

最新推荐文章于 2024-07-12 15:40:04 发布

蒂莫韦尔纳

最新推荐文章于 2024-07-12 15:40:04 发布

阅读量1w

点赞数 14

分类专栏：深度学习文章标签： keras 深度学习 GPU tensorflow CUDA

本文链接：https://blog.csdn.net/qq_32728345/article/details/81570504

版权

本文介绍了如何在Windows 10系统上配置支持GPU加速的Keras深度学习环境，包括CUDA和cuDNN的安装、Anaconda配置、tensorflow-gpu与keras的安装，最后通过简单的测试验证GPU运行成功。

摘要由CSDN通过智能技术生成

本人目前在用深度学习方法做命名实体识别。起初学习阶段因为数据量不大，所以直接用CPU来训练模型；不过真正落实到研究则需要大量数据才能训练出泛化能力强的模型，这时就要利用GPU来加速神经网络训练。

本人通过查阅资料配置好了GPU加速版的Keras框架，下面分享给大家。

首先在确定你的显卡是否支持CUDA，可以在这里查看，如果显卡在列表中则说明支持CUDA。
然后查看显卡支持的CUDA版本，方法是进入NVIDIA控制面板==>帮助==>系统信息==>组件==>NVCUDA.DLL==>产品名称。这里我的显卡支持的CUDA版本为9.2，其他版本我尝试过，都无法成功安装。所以要安装显卡支持的CUDA版本，才能进行后续的配置。
CUDA下载地址 https://developer.nvidia.com/cuda-toolkit-arc

关注