g4dn.xlarge T4 GPU - kernel dead when loading a 7b LLM

0

I am loading the mistral-7b model from HF Transformers, however, I keep getting kernel dead error on jupyter notebook.

from transformers import AutoModelForSequenceClassification, AutoModelForCausalLM

model = AutoModelForCausalLM.from_pretrained("mistralai/Mistral-7B-v0.1")
model

I am trying to inspect the jupyter.log stream in cloudwatch, however, I do not see any errors specific to the kernels. Where should I be looking for the dead kernel error?

kev
質問済み 2ヶ月前159ビュー
1回答
0

When you see "The kernel appears to have died" message and "Dead kernel" label, please try to run the same code in a Terminal session instead of in a Notebook. Then you could see the error message of the root cause. (Most likely the python process died due to native module crash). You would have to copy & paste Python code cells one by one.

AWS
回答済み 1ヶ月前

ログインしていません。 ログイン 回答を投稿する。

優れた回答とは、質問に明確に答え、建設的なフィードバックを提供し、質問者の専門分野におけるスキルの向上を促すものです。

質問に答えるためのガイドライン

関連するコンテンツ