Merge pull request #296 from ilisin/master

add GPU support on Mac
datamllab · Jul 11, 2023 · 7fc56ed · 7fc56ed
2 parents a093f51 + 05cc499
commit 7fc56ed
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/rlcard/utils/utils.py b/rlcard/utils/utils.py
@@ -19,7 +19,10 @@ def set_seed(seed):
 
 def get_device():
     import torch
-    if torch.cuda.is_available():
+    if torch.backends.mps.is_available():
+        device = torch.device("mps:0")
+        print("--> Running on the GPU")
+    elif torch.cuda.is_available():
         device = torch.device("cuda:0")
         print("--> Running on the GPU")
     else: