diff --git a/invesalius/gui/brain_seg_dialog.py b/invesalius/gui/brain_seg_dialog.py
index 577c92f..ea725d5 100644
--- a/invesalius/gui/brain_seg_dialog.py
+++ b/invesalius/gui/brain_seg_dialog.py
@@ -21,6 +21,7 @@ from invesalius.gui import dialogs
 HAS_THEANO = bool(importlib.util.find_spec("theano"))
 HAS_PLAIDML = bool(importlib.util.find_spec("plaidml"))
 PLAIDML_DEVICES = {}
+TORCH_DEVICES = {}
 
 try:
     import torch
--
libgit2 0.21.2