amd没有cuda版本,但amd显卡可以通过安装rocm使用tensorflow和pytorch rocm官网: https://www.amd.com/zh-hans/graphics/servers-solutions-rocm-ml