0

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 上运行
4

1 回答 1

0

谢谢克里斯派索_

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

于 2021-12-09T07:58:45.310 回答