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

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


動画版(生成AIによる自動生成): この記事の内容をずんだもん×四国めたんの掛け合いで解説しています。自動生成のため、内容に誤りがある可能性があります。正確な情報は記事本文をご参照ください。