added mps code and comment

Author: alvaropp

beginner_source/basics/buildmodel_tutorial.py

@@ -32,12 +32,11 @@
 #############################################
 # Get Device for Training
 # -----------------------
-# We want to be able to train our model on a hardware accelerator like the GPU,
-# if it is available. Let's check to see if
-# `torch.cuda <https://pytorch.org/docs/stable/notes/cuda.html>`_ is available, else we
-# continue to use the CPU.
+# We want to be able to train our model on a hardware accelerator like the GPU or MPS,
+# if available. Let's check to see if `torch.cuda <https://pytorch.org/docs/stable/notes/cuda.html>`_
+# or `torch.backends.mps <https://pytorch.org/docs/stable/notes/mps.html> are available, otherwise we use the CPU.
 
-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")
 
 ##############################################

beginner_source/basics/quickstart_tutorial.py

@@ -84,9 +84,9 @@
 # To define a neural network in PyTorch, we create a class that inherits
 # from `nn.Module <https://pytorch.org/docs/stable/generated/torch.nn.Module.html>`_. We define the layers of the network
 # in the ``__init__`` function and specify how data will pass through the network in the ``forward`` function. To accelerate
-# operations in the neural network, we move it to the GPU if available.
+# operations in the neural network, we move it to the GPU or MPS if available.
 
-# Get cpu or gpu device for training.
+# Get cpu, gpu or mps device for training.
 device = "cuda" if torch.cuda.is_available() else "mps" if torch.backends.mps.is_available() else "cpu"
 print(f"Using {device} device")