diff --git a/setup.py b/setup.py
index 0c156fa..72bfe90 100644
--- a/setup.py
+++ b/setup.py
@@ -17,15 +17,16 @@
         'pytest==7.1.2',
         'pytest-cov==3.0.0',
         'pytorch_lightning==1.8.3',
-        'pytype==2023.1.17',
+        'pytype==2024.2.27',
         'sphinx==7.2.6',
         'sphinx-autoapi==3.0.0',
         'sphinx-rtd-theme==1.3.0',
         'sphinxcontrib-napoleon==0.7',
         'sphinxcontrib-katex==0.9.9',
+        'tensorboard==2.16.2',
         'tqdm==4.64.0',
-        'torch==1.12.0',
-        'torchvision==0.13.0',
+        'torch==2.2.1',
+        'torchvision==0.17.1',
     ],
     entry_points={
         'console_scripts': [