Keras使用GPU训练

TensorFlow、Keras与GPU之间的版本对应

  • 版本问题—keras和tensorflow的版本对应关系
  • 环境部署中cuda对应的tensorflow-gpu、keras版本、pytorch的对应版本
  • 官方文档中的版本对应

使用GPU训练Keras模型

  • Keras——检查GPU是否可用
  • 如何使用GPU训练keras模型

查看显卡驱动版本与CUDA Tookit版本对应

先在自己电脑的NVIDIA控制面板中查看显卡驱动程序版本
帮助 – 系统信息
Keras使用GPU训练_第1张图片
然后去官网中查看对应CUDA Tookit版本,点此跳转
Keras使用GPU训练_第2张图片
对照显卡程序版本460.79,应该安装CUDA11.3版本

参考:

  • 手把手教你:Windows系统上安装GPU深度学习环境
  • 根据显卡驱动版本安装对应版本的CUDAToolkit

你可能感兴趣的:(深度学习,tensorflow,深度学习)