linux下ROCm HIP,和CUDA代码相似，还是需要移植。tf和pytorch有ROCm下开源的移植。windows下有tensorflow-directml，tensorflow2 directml 插件, pytorch directml，所有这些比cuda版都少一些高级功能。

【 在 yzjba 的大作中提到: 】
: AMD也搞了GPU计算吗，网上基于CUDA的代码能直接用吗
--
修改:netvideo FROM 125.69.13.*
FROM 125.69.13.*