Kernel dying when using scvi to train a model (m1 mac)

Hi, I’m trying to learning analysis by following various tutorials and when using scvi to predict doublets, my kernel keeps dying.

This is the code that I’m trying to run

vae = scvi.model.SCVI(adata)

There was post about this same issue and two suggestions were to ensure sufficient RAM and that pytorch is working.

I’m assuming 64gb of RAM is sufficient.

However, I’m not exactly sure how to check if pytorch is working. My environment only contains pytorch-lightning. Do I just need to install pytorch or follow one of the tutorials on getting pytorch to work with m1 macs?

Thank you!