如何使用 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 中，我正在尝试使用 transformmer-cli 将 bioBERT 权重转换为 PyTorch。我收到“无法加载动态库 'cudart64_110.dll'”的警告。这会影响以后的性能吗？
2. 第二个问题：在单元格 9 中，我的模型加载非常慢，因为它只使用 CPU。如何在我的 AMD GPU
上将模型设置为 运行

Answer 1

感谢chrispresso

AMD ROCm 似乎是要走的路，但在 linux

下需要 运行