Lab setup failure: PyTorch not installed?

[Cindy_Yuan]

My lab setup failed at Step 3 loading LLM model:

from datasets import load_dataset
from transformers import AutoModelForSeq2SeqLM
from transformers import AutoTokenizer
from transformers import GenerationConfig

I got the following error:

None of PyTorch, TensorFlow >= 2.0, or Flax have been found. Models won’t be available and only tokenizers, configuration and file/data utilities can be used.

How do I fix it?

[Cindy_Yuan]

I believe this was the previous install of pytorch failed. Here is a screenshot showing the output:

[Cindy_Yuan]

Never mind. I cleared my workspace and restarted from the beginning, and this time it passed without a problem.