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 安装要求:
- N卡架构为图灵架构或以下。
- 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中主流深度学习框架安装细节的主要内容,如果未能解决你的问题,请参考以下文章