如何用AMD运行Py火炬Bert

Question

github代码：https://github.com/bellowman/Deep-Learning-Practice/blob/main/BioBert%20for%20Multi%20Label%20AMD.ipynb

大家好,

我是个生手，tensorflow和BERT。我家里有一台带有AMD Ryzen 7 1800 x和Radeon RX 6600显卡的机器。

我正试图在家里运行一个bioBERT模型。我很难利用我的模型使用我的AMD卡。我发了我的github笔记本。我在3号和9号牢房有麻烦。

1. 第一个问题:在单元格3中，我正试图用变形器-cli将bioBERT的权重转换为PyTorch。我收到了“无法加载动态库'cudart64_110.dll'”的警告。这会影响以后的表现吗？
2. 第二个问题:在单元格9中，我的模型负载非常慢，因为它只使用CPU。怎样才能让模型在我的AMD GPU上运行？

Answer 1

谢谢意式咖啡

AMD ROCm似乎是可行的，但它需要在linux下运行。