2023年了,AMD显卡做深度学习怎样

总结

个人认为Nvidia显卡目前依然是比较好的方案,使用AMD显卡需要针对性的看具体用途并做测试,原因如下:

1 CUDA具备先发优势,开源生态成熟,可能在很长一段时间内都是主流,其他的追随者虽然提供类似的加速计算工具,很难得到广泛的开源生态支持。

2 其他平台的底层API,很难做到和CUDA接口完全一致,目前ROCm模仿的是英伟达9和10系列的显卡的计算接口,可能还需要一些时间积累。

3 其他平台的稳定性也有所不足(可能导致人工开发维护成本增加),CUDA经过很多年的积累,很多开源社区的使用也让其发现和修复了很多bug,不只是计算平台本身,可能还涉及系统、硬件等多方面,在没有足够的开源生态去帮助调试发现bug的时候,计算平台厂商的工作量是非常大的。

4 成本优势相比维护成本微弱,虚拟币挖矿的大潮过去后,显卡价格落差在缩小,游戏性能相近的RTX 4080和RX7900XT,售价分别是8000和6000左右,并且价格方面可以通过二手显卡、低配显卡方面妥协,并且A卡很可能速度上也有牺牲,游戏性能相当的卡可能跑AI就会有一定的差距)

5 除了软件生态不足,A卡连自家的显卡目前支持度还非常有限,消费级显卡只有两款支持,目前最新的显卡RX7900XT都不支持自己的计算平台ROCm

名词解释

ROCm:AMD发布的GPU加速计算库,只支持AMD显卡,支持linux,据说windows也开始支持

DirectML:微软发布GPU加速计算库,据说支持AMD、NVIDIA等显卡,只支持windows

CUDA:英伟达发布的GPU加速计算库,只支持NVIDIA显卡,支持windows和linux,下游软件生态最成熟稳定


三者横向对比参考资料:

有人使用过Tensorflow-DirectML嘛,性能和cuda或者rocm相比如何?

方案介绍

方案一Linux+ROCm

优点:

1性价比略高

2 Pytorch2.0兼容

缺点:

1软件生态不成熟,AMD早在2016年4月就发布了ROCm平台,迄今已经整整7年,但因为起步太晚,而且官方支持力度有限,尽管开源,却一直没能发展开来。

2部分底层接口尚未实现,还不能完全替代Linux+CUDA方案。

3目前支持的硬件非常有限,目前消费级显卡仅仅支持老款的RX6900XT、RX6600、VII

方案二Windows+ROCm

参考链接:

7年了!AMD CUDA杀手ROCm终于登陆Windows、支持游戏显卡

优点:

暂无

缺点:

1下游软件生态不成熟,目前所有基于pytorch的模型(如chatGLM)都无法用这种方式运行。虽然在今年4月14日,AMD宣布ROCm宣布支持windows,但是截止到6月25日,pytorch官方网站显示windows系统下pytorch还不支持ROCm,需要持续关注

2支持的显卡非常有限,目前windows下只支持Radeon Pro W6800、RX 6900 XT、RX 6600

方案三Windows+DirectML

参考链接

AMD GPU,PyTorch,Windows和DirectML

Enable PyTorch with DirectML on Windows

优点:

1支持多种GPU

缺点:

1软件支持十分匮乏,目前pytorch已经更新到2.0,目前官方文档描述DirectML只支持pytorch1.13,仅仅就chatGLM来说,官方文档中要求pytorch版本大于1.10,如果只是运行chatGLM的推理,可能够用(具体稳定性如何,是否支持还需要进一步真机测试),只是可能未来程序升级可能会比较受限,比如用其他的更新的模型,对版本要求可能会更高。


方案四Windows+CPU

优点:

1不借助显卡加速,只用CPU计算,部分模型在CPU和GPU做推理速度差别不明显,多数模型速度虽有明显差异,但也能将就使用。

2硬件成本更低,必要的情况下可升级显卡加快速度。

3软件支持兼容性好,几乎所有模型都支持CPU推理。

缺点:

1可以做推理,但是不适合做模型训练。

方案五Windows+CUDA

优点:

1基本上绝大多数的Nvidia显卡都支持

2追求性价比可以考虑二手显卡、低配显卡

缺点:

1相比linux+CUDA,开源生态略显不足,程序稳定性略低


方案六Linux+CUDA

优点:

1软件生态完善,最活跃,目前流行的多卡并行计算的开源框架多数都是基于该方案

2基本上绝大多数的Nvidia显卡都支持

3追求性价比可以考虑二手显卡、低配显卡

4稳定性最高,减少后期维护成本

缺点:

暂无

©著作权归作者所有,转载或内容合作请联系作者
  • 序言:七十年代末,一起剥皮案震惊了整个滨河市,随后出现的几起案子,更是在滨河造成了极大的恐慌,老刑警刘岩,带你破解...
    沈念sama阅读 204,921评论 6 478
  • 序言:滨河连续发生了三起死亡事件,死亡现场离奇诡异,居然都是意外死亡,警方通过查阅死者的电脑和手机,发现死者居然都...
    沈念sama阅读 87,635评论 2 381
  • 文/潘晓璐 我一进店门,熙熙楼的掌柜王于贵愁眉苦脸地迎上来,“玉大人,你说我怎么就摊上这事。” “怎么了?”我有些...
    开封第一讲书人阅读 151,393评论 0 338
  • 文/不坏的土叔 我叫张陵,是天一观的道长。 经常有香客问我,道长,这世上最难降的妖魔是什么? 我笑而不...
    开封第一讲书人阅读 54,836评论 1 277
  • 正文 为了忘掉前任,我火速办了婚礼,结果婚礼上,老公的妹妹穿的比我还像新娘。我一直安慰自己,他们只是感情好,可当我...
    茶点故事阅读 63,833评论 5 368
  • 文/花漫 我一把揭开白布。 她就那样静静地躺着,像睡着了一般。 火红的嫁衣衬着肌肤如雪。 梳的纹丝不乱的头发上,一...
    开封第一讲书人阅读 48,685评论 1 281
  • 那天,我揣着相机与录音,去河边找鬼。 笑死,一个胖子当着我的面吹牛,可吹牛的内容都是我干的。 我是一名探鬼主播,决...
    沈念sama阅读 38,043评论 3 399
  • 文/苍兰香墨 我猛地睁开眼,长吁一口气:“原来是场噩梦啊……” “哼!你这毒妇竟也来了?” 一声冷哼从身侧响起,我...
    开封第一讲书人阅读 36,694评论 0 258
  • 序言:老挝万荣一对情侣失踪,失踪者是张志新(化名)和其女友刘颖,没想到半个月后,有当地人在树林里发现了一具尸体,经...
    沈念sama阅读 42,671评论 1 300
  • 正文 独居荒郊野岭守林人离奇死亡,尸身上长有42处带血的脓包…… 初始之章·张勋 以下内容为张勋视角 年9月15日...
    茶点故事阅读 35,670评论 2 321
  • 正文 我和宋清朗相恋三年,在试婚纱的时候发现自己被绿了。 大学时的朋友给我发了我未婚夫和他白月光在一起吃饭的照片。...
    茶点故事阅读 37,779评论 1 332
  • 序言:一个原本活蹦乱跳的男人离奇死亡,死状恐怖,灵堂内的尸体忽然破棺而出,到底是诈尸还是另有隐情,我是刑警宁泽,带...
    沈念sama阅读 33,424评论 4 321
  • 正文 年R本政府宣布,位于F岛的核电站,受9级特大地震影响,放射性物质发生泄漏。R本人自食恶果不足惜,却给世界环境...
    茶点故事阅读 39,027评论 3 307
  • 文/蒙蒙 一、第九天 我趴在偏房一处隐蔽的房顶上张望。 院中可真热闹,春花似锦、人声如沸。这庄子的主人今日做“春日...
    开封第一讲书人阅读 29,984评论 0 19
  • 文/苍兰香墨 我抬头看了看天上的太阳。三九已至,却和暖如春,着一层夹袄步出监牢的瞬间,已是汗流浃背。 一阵脚步声响...
    开封第一讲书人阅读 31,214评论 1 260
  • 我被黑心中介骗来泰国打工, 没想到刚下飞机就差点儿被人妖公主榨干…… 1. 我叫王不留,地道东北人。 一个月前我还...
    沈念sama阅读 45,108评论 2 351
  • 正文 我出身青楼,却偏偏与公主长得像,于是被迫代替她去往敌国和亲。 传闻我的和亲对象是个残疾皇子,可洞房花烛夜当晚...
    茶点故事阅读 42,517评论 2 343

推荐阅读更多精彩内容