diff --git a/setup.py b/setup.py
index cb3dfdb..f890462 100644
--- a/setup.py
+++ b/setup.py
@@ -14,7 +14,7 @@
     long_description_content_type="text/markdown",
     url="https://github.com/microsoft/ProteinFunctions", #TODO: Change repo name
     install_requires=[
-        "torch==2.2.0",  # PyTorch
+        "torch==2.6.0",  # PyTorch
         "torchvision==0.15.2",  # TorchVision
         # "pytorch-cuda==11.8", #PyTorch CUDA
         "pandas==1.5.2",  # Pandas