深度学习GPU显卡选择
今天我导问我什么显卡做深度学习好一点,我觉得还是调查才有发言权。
A卡orN卡A卡指AMD的显卡,N卡是指英伟达的显卡。由于做深度学习需要使用CUDA,所以只能使用N卡。当然目前主流的深度学习框架如Pytorch、TensorFlow等都支持安装cpu版本,没有显卡其实也能玩。
算力对比对于个人使用而言,消费级的显卡就够用了,如现在比较新的RTX3060等等显卡。对比可用在服务器上的显卡而言,此类显卡单卡性能已经相当不错了,只是功耗较高且不便于扩展,个人做一些测试小训练啥的都没有问题。
用做深度学习训练时,显存买越大的越好,这样可以支持更大的batchsize。
需要注意GTX系列大于1080ti的显卡都没有tensorcore,不建议用来做深度学习(当然用也是可以用的,毕竟可以安装CUDA)。
具体的显卡选择可以在英伟达官网查看,算力对比。以下截图一些常见显卡,貌似算力大于5.0的就比较推荐用来跑神经网络了。
服务器端常用显卡:
消费级显卡: