Windows中主流深度学习框架安装细节

Posted Mister Meng

tags:

篇首语:本文由小常识网(cha138.com)小编为大家整理,主要介绍了Windows中主流深度学习框架安装细节相关的知识,希望对你有一定的参考价值。

1. 硬件、cuda、框架

最佳配置:硬件配置中的cuda版本号=当前安装cuda版本号=框架要求的cuda版本号。

向下兼容:硬件配置中的cuda版本号>当前安装cuda版本号>框架要求的cuda版本号。

2. Tensorflow 1.x

目前,pytorch和tensorflow 2.x向下兼容做得很好,易安装。Tensorflow 1.x 安装要求:

  1. N卡架构为图灵架构或以下。
  2. python 3.7 + cudatoolkit 10.0 + cudnn 7.6.5, tensorflow 1.14 + keras 2.3.1 + h5py 2.10 + protobuf 3.19.0 

以上是关于Windows中主流深度学习框架安装细节的主要内容,如果未能解决你的问题,请参考以下文章

深度学习及机器学习框架对比摘要

TensorFlow与主流深度学习框架对比

主流的深度学习框架基本知识

主流深度学习框架对比(TensorFlowKerasMXNetPyTorch)

五大主流深度学习框架对比:MXNET是最好选择

主流深度学习框架对比