Add support of mps for Mac M1 CPUs (#2165)

Author: zshao

beginner_source/basics/quickstart_tutorial.py

@@ -87,7 +87,7 @@
 # operations in the neural network, we move it to the GPU if available.
 
 # Get cpu or gpu device for training.
-device = "cuda" if torch.cuda.is_available() else "cpu"
+device = "cuda" if torch.cuda.is_available() else "mps" if torch.backends.mps.is_available() else "cpu"
 print(f"Using {device} device")
 
 # Define model