如何用AMD运行Pytorch Bert

Question

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

大家好，

我是pytorch、tensorflow和BERT的初学者。我家里有一台机器，配有AMD Ryzen 7 1800x和Radeon RX 6600视频卡。

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

1. 第一个问题：在细胞3中，我试图用转化子cli将bioBERT重量转换为PyTorch。我收到了"；无法加载动态库"cudart64_110.dll"；。这会影响以后的性能吗
2. 第二个问题：在单元9中，我的模型加载非常慢，因为它只使用了CPU。我如何让模型在我的AMD GPU上运行

Answer 1

感谢chrispresso

AMD ROCm似乎是一条路要走，但它需要一个在linux 下运行