diff --git a/requirements.txt b/requirements.txt
index b466db91a7ff7c31c1132167f067e5e51e1fa424..26d4c2843296d39a9821ac66b7fc3f2fb5f93e8f 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -11,6 +11,12 @@ PyYAML>=5.3.1        # YAML parser and emitter
 pandas==2.0.3        # Data manipulation and analysis
 tqdm>=4.64.0         # Progress bar library
 
+# Torch -----------------------------------------------------------------------
+
+torch==2.0.0        
+torchvision==0.15.1
+torchaudio==2.0.1
+
 # Plotting --------------------------------------------------------------------
 
 seaborn>=0.11.0      # Statistical data visualization