如何选一块支持深度学习的Nvidia显卡

本文以tensorflow为例,具体讲解什么样的nvidia显卡可以支持深度学习
nvidia显卡多如牛毛,但并不是所有的型号都可以支持tensorflow的所有型号,本文讲解如何确定一个具体型号的显卡可以支持哪个tensorflow版本

1.在nvidia官网搜索显卡对应最高驱动版本
驱动程序下载
本文以660为例,660支持驱动最高版本为446.14

2.在驱动下载页面的‘其他信息’内,点击Game Ready Driver Release Notes (v446.14),打开Release Notes

3.在Release Notes中Software Module Versions章节就可以找到CUDA - 10.2 ,故该显卡最高支持CUDA - 10.2 版本,对应tensorflow的tensorflow_gpu-2.0.0版本

CUDA版本和驱动版本对应关系:
在这里插入图片描述
tensorflow与CUDA版本对应关系
在这里插入图片描述

Logo

CSDN联合极客时间,共同打造面向开发者的精品内容学习社区,助力成长!

更多推荐