- 主题:AMD tensorflow-directml
AMD也搞了GPU计算吗，网上基于CUDA的代码能直接用吗
【 在 netvideo 的大作中提到: 】
: tensorflow-directml
: 红米6800H笔记本核显速度约等于 gtx1070 的 45%，gpu温度不到60度。RX7900XT速度约等于gtx1070的3倍速。
--
FROM 117.136.0.*
那还是没办法取代啊，必须做到代码一致才能替代
【 在 netvideo 的大作中提到: 】
: linux下ROCm HIP,和CUDA代码相似，还是需要移植。tf和pytorch有ROCm下开源的移植。windows下有tensorflow-directml，tensorflow2 directml 插件, pytorch directml，所有这些比cuda版都少一些高级功能。
:
--
FROM 117.136.0.*
那还不错啊，可以试试
【 在 netvideo 的大作中提到: 】
: tensorflow和pytorch的代码大多数情况可以直接用。
:
--
FROM 117.136.0.*
可惜没有ubuntu版本
【 在 netvideo 的大作中提到: 】
: directml的tf,pytorch还支持intel显卡，支持DirectX12就可以。所以在程序部署上有一些独特的优势。
:
--
FROM 117.136.0.*
现在好多开发都是linux下的，做windows意义不大
【 在 netvideo 的大作中提到: 】
: linux就安心用cuda，除非没n卡可用。
: windows的directml版好在安装方便，同时支持多家显卡包括核显，有一定意义。
:
--
FROM 117.136.0.*