問題タブ [amd-gpu]

For questions regarding programming in ECMAScript (JavaScript/JS) and its various dialects/implementations (excluding ActionScript). Note JavaScript is NOT the same as Java! Please include all relevant tags on your question; e.g., [node.js], [jquery], [json], [reactjs], [angular], [ember.js], [vue.js], [typescript], [svelte], etc.

0 投票する
1 に答える
136 参照

python - AMD で Pytorch Bert を実行する方法

github コード: https://github.com/belowman/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. 2 番目の質問: セル 9 では、CPU のみを使用しているため、モデルの読み込みが非常に遅いです。モデルを AMD GPU で実行するにはどうすればよいですか