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?

asked a year ago299 views
1 Answer
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
answered a year ago

You are not logged in. Log in to post an answer.

A good answer clearly answers the question and provides constructive feedback and encourages professional growth in the question asker.

Guidelines for Answering Questions