AMD显卡使用Pytorch

说明

不要瞎折腾，AMD显卡使用Pytorch必须需要在Linux上，windows上无法通过rcom实现调用显卡，或许可以通过docker(但是在22年测试的时候是不可以的),微软的pytorch-directml是可以调用显卡的在windows上。

我已经在6700xt上测试过了，然而当遇到cuda时候，你依然是没有办法的，总不可能改代码吧。总之代码兼容性很差

这有一篇

在windows上通过pytorch-directml利用AMD显卡加速stable-diffusion

https://zhuanlan.zhihu.com/p/601903035

这有一篇是AMD显卡使用Pytorch深度学习配置(RX6600 ubuntu20.04为例)-2022.9.20

https://zhuanlan.zhihu.com/p/566112395

tensorflow可能在amd上运行的更好点，但是建议深度学习还是n卡

https://tensorflow.google.cn

或者使用谷歌的colab 地址:https://colab.research.google.com/

可以看一下这个文章

https://www.cnblogs.com/runrab/p/17218957.html