From 5a80d96c52f7fa2c0efedbcef797ddd5de9bfd1b Mon Sep 17 00:00:00 2001 From: Zheng Shao Date: Fri, 6 Jan 2023 10:04:29 -0800 Subject: [PATCH] Add support of mps for Mac M1 CPUs --- beginner_source/basics/quickstart_tutorial.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/beginner_source/basics/quickstart_tutorial.py b/beginner_source/basics/quickstart_tutorial.py index 75f89a1090e..8cba94bcc04 100644 --- a/beginner_source/basics/quickstart_tutorial.py +++ b/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