Added one liner to set device pytorch

AnupJoseph · Jul 22, 2023 · 2ae1603 · 2ae1603
1 parent 8e1845c
commit 2ae1603
Show file tree

Hide file tree

Showing 3 changed files with 27 additions and 1 deletion.
diff --git a/config/_default/languages.toml → config/_default/_languages.toml b/config/_default/languages.toml → config/_default/_languages.toml
diff --git a/config/_default/config.toml b/config/_default/config.toml
@@ -6,7 +6,7 @@ title = "Utils"
 
 # Theme i18n support
 # Available values: en, fr, id, ja, ko, pt-br, zh-cn, zh-tw, es, de, nl, it, th, el, uk, ar
-DefaultContentLanguage = "en"
+defaultContentLanguage = "en"
 
 # Set hasCJKLanguage to true if DefaultContentLanguage is in [zh-cn ja ko]
 # This will make .Summary and .WordCount behave correctly for CJK languages.

diff --git a/content/post/device_oneliner/index.md b/content/post/device_oneliner/index.md
@@ -0,0 +1,26 @@
+---
+title: Setting device in Torch
+description: Simplest method to have device specific Pytorch code
+slug: device_oneline
+date: 22-07-2023 00:00:00+0000
+categories:
+    - Python
+    - Pytorch
+tags:
+    - python
+    - torch
+---
+
+A simple onliner to set device in PyTorch. 
+```python
+import torch
+
+device = (
+    "cuda"
+    if torch.cuda.is_available()
+    else "mps"
+    if torch.backends.mps.is_available()
+    else "cpu"
+)
+print(f"Using {device} device")
+```