基于tensorflow-gpu-2.6.0 + Intel CPU + GTX1660 SUPER 使用CUDA计算

tensorflow-gpu-2.6.0 + Intel CPU + GTX1660 SUPER 使用CUDA计算

前言:
本文为基于NVIDIA GPU 进行 tensorflow计算的配置说明,如果电脑中安装有cuDNN 且Tensorflow支持的话,会默认优先使用GPU进行计算,如果配置GPU失败,会使用CPU计算
关于计算能力,测试了10万张级别的【图片识别分类】,GTX 1660 SUPER GPU的计算时间大概会比I5-7500 缩短8-10倍。

# 硬件 模型计算完成时间
1 GTX 1660 SUPER ≈ 25 分钟
2 I5-7500 ≈ 7.5 - 9 小时
# 软件 说明 查询地址
1 cuda显卡 在cuda-gpus列表查询当前显卡是否支持cuda功能及Compute Capability版本 cuda-gpus
2 CUDA 版本 通过查询Tensorflow-GPU版本列表,当前文档中最高支持CUDA 11.2 Tensorflow-GPU
3 显卡驱动 不需

你可能感兴趣的:(环境构建,Windows,tensorflow,python,深度学习)