diff --git a/pyproject.toml b/pyproject.toml
index d6e58a4..bada2a6 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -4,7 +4,7 @@ build-backend = "hatchling.build"
 
 [project]
 name = "soup-cli"
-version = "0.2.0"
+version = "0.2.1"
 description = "Fine-tune LLMs in one command. No SSH, no config hell."
 readme = "README.md"
 license = "MIT"
diff --git a/soup_cli/__init__.py b/soup_cli/__init__.py
index 1f58073..37d1225 100644
--- a/soup_cli/__init__.py
+++ b/soup_cli/__init__.py
@@ -1,3 +1,3 @@
 """Soup CLI — Fine-tune LLMs in one command."""
 
-__version__ = "0.2.0"
+__version__ = "0.2.1"
diff --git a/soup_cli/monitoring/callback.py b/soup_cli/monitoring/callback.py
index 8af6df6..1fc0ab2 100644
--- a/soup_cli/monitoring/callback.py
+++ b/soup_cli/monitoring/callback.py
@@ -47,7 +47,7 @@ class SoupTrainerCallback(TrainerCallback):
 
             if torch.cuda.is_available():
                 used = torch.cuda.memory_allocated() / (1024**3)
-                total = torch.cuda.get_device_properties(0).total_mem / (1024**3)
+                total = torch.cuda.get_device_properties(0).total_memory / (1024**3)
                 gpu_mem = f"{used:.1f}/{total:.1f} GB"
         except Exception:
             pass
diff --git a/soup_cli/utils/gpu.py b/soup_cli/utils/gpu.py
index 929feba..210d970 100644
--- a/soup_cli/utils/gpu.py
+++ b/soup_cli/utils/gpu.py
@@ -25,7 +25,7 @@ def get_gpu_info() -> dict:
         import torch
 
         if torch.cuda.is_available():
-            total = torch.cuda.get_device_properties(0).total_mem
+            total = torch.cuda.get_device_properties(0).total_memory
             total_gb = total / (1024**3)
             return {
                 "memory_total": f"{total_gb:.1f} GB",