Windows 下 AMD显卡训练模型有救了:pytorch_directml 下运行Transformers
Windows下amd显卡训练transformer模型。安装方法参见:Windows下用amd显卡训练:Pytorch-directml重大升级,改为pytorch插件形式,兼容更好_amd显卡pytorch_znsoft的博客-CSDN博客importosimportimptry:imp.find_module('torch_directml')found_directml=Trueimpor