tensorflow在AMD上用ROCm框架实现GPU训练(亲测有效)

众所周知,a卡对于tensorflow gpu的支持很小,要想使用的话通常都得配n卡的电脑。而我只是感兴趣一个项目,想要跑出来一个模型,若是不用gpu加速,可能得跑一个月,因此不得不寻找在a卡上的解决方案。好在,终于找到了一个解决方案,虽然只是取巧,速度也不如正常n卡跑的快,但已经很不错了。html 个人配置: 显卡:RX 550 所用系统:虚拟机ubuntu 18 注:若是是虚拟机的话,最好分配
相关文章
相关标签/搜索