🙆‍♀️

Vertex AIのworkbenchを使用した際、HuggingFaceのTrainer()が開始されない事象への対処法

2022/11/21に公開

Google CloudのVertex AIのworkbenchを使用した際、HuggingFaceのTrainer()が開始されない事象に遭遇しました。

この件について、以下のページで同様の不具合が報告されていました。

https://stackoverflow.com/questions/73415068/huggingface-trainer-does-nothing-only-on-vertex-ai-workbench-works-on-colab

当初、以下のような「PyTorch」の環境を選んでいましたが、この環境で上記の不具合が生じました。

そこで、上記の記事にある通り、「Python」の環境を選択することで、上記の不具合を避けることができました。

なお、上記の環境を選んだ場合、まず以下のような実行が必要です。

conda install pytorch cudatoolkit=11.0 -c pytorch

同様のことでお困りの方の参考になりましたら幸いです。

Discussion