CUDA版本: 11.8,但由于本人电脑无英伟达显卡,这是在服务器上训练时直接拉的社区镜像里的配置,具体兼容性还请自行搜索。 吐槽下Linux下是装CUDA是真的难搞,之前想换下CUDA版本搞一天都没搞定,最后直接拉大神镜像了 AMD请安装 tensorflow-directml-plugin 参考 知乎 ...