深度学习GPU显卡选择

4,967 阅读1分钟

今天我导问我什么显卡做深度学习好一点,我觉得还是调查才有发言权。

A卡 or N卡

A卡指AMD的显卡,N卡是指英伟达的显卡。由于做深度学习需要使用CUDA,所以只能使用N卡。当然目前主流的深度学习框架如Pytorch、TensorFlow等都支持安装cpu版本,没有显卡其实也能玩。

算力对比

对于个人使用而言,消费级的显卡就够用了,如现在比较新的RTX3060等等显卡。对比可用在服务器上的显卡而言,此类显卡单卡性能已经相当不错了,只是功耗较高且不便于扩展,个人做一些测试小训练啥的都没有问题。

用做深度学习训练时,显存买越大的越好,这样可以支持更大的batch size。

需要注意GTX系列大于1080ti的显卡都没有tensor core,不建议用来做深度学习(当然用也是可以用的,毕竟可以安装CUDA)。

具体的显卡选择可以在英伟达官网查看,算力对比。以下截图一些常见显卡,貌似算力大于5.0的就比较推荐用来跑神经网络了。

服务器端常用显卡:

image.png

消费级显卡:

image.png