【发布时间】：2022-01-13 16:29:53
【问题描述】：
大家好，
我是 pytorch、tensorflow 和 BERT 的初学者。我家里有一台装有 AMD Ryzen 7 1800x 和 Radeon RX 6600 显卡的机器。
我正在尝试在家中运行 bioBERT 模型。我无法利用我的模型来使用我的 AMD 卡。我发布了我的 github 笔记本。我在 3 号和 9 号牢房有问题。
- 第一个问题：在单元格 3 中，我正在尝试使用 transformmer-cli 将 bioBERT 权重转换为 PyTorch。我收到“无法加载动态库'cudart64_110.dll'”的警告。这会影响以后的性能吗？
- 第二个问题：在单元格 9 中，我的模型加载速度非常慢，因为它只使用 CPU。如何让模型在我的 AMD GPU 上运行
【问题讨论】：
-
好吧，您无法加载 cudart dll，因为您没有合适的显卡。要将 PyTorch 与 AMD 一起使用，您需要关注 this。另一种选择是仅使用 google colab 并加载该 ipynb，然后您就不会遇到这些问题了。
-
是的，我对 Google Colab 很熟悉，但我确实想让它与 AMD 一起工作。想知道是否有人知道确切的代码更改或使其与 AMD 一起使用的步骤
-
你是否安装了带有 ROC-m 包的 PyTorch？单元格中的代码可能不是问题，很可能是您安装了错误的 PyTorch。
-
啊，看来我必须在 linux 上执行此操作才能使其正常工作。谢谢你的建议。
标签： python deep-learning pytorch bert-language-model amd-gpu