Possible bug in recognizing mps
accelerator even though PyTorch seems to register the mps
device?
#19868
Labels
mps
accelerator even though PyTorch seems to register the mps
device?
#19868
Bug description
Hi, I'm a new user to pl, so apologies if this ends up not being a bug.
I am running PyTorch and pytorch-lightning within a Jupyter notebook on my MacOSX M1. My conda environment is from
miniforge
, which supports the Arm64 distro. As expected, the following print statements show thatmps
is available.However, when I transition to initializing the
Trainer
:I get an error shown below.
What version are you seeing the problem on?
v2.2
How to reproduce the bug
Error messages and logs
Environment
Current environment
More info
I looked at the following link, but I believe I am in the right conda environment, so this led to a dead end for me at least.
The text was updated successfully, but these errors were encountered: