Commit 35fdb1a8 authored by Kelly Guo's avatar Kelly Guo Committed by Kelly Guo

Updates torch to 2.5.1 (#168)

# Description
Updates torch to 2.5.1
parent 16c38492
...@@ -19,7 +19,7 @@ EXTENSION_TOML_DATA = toml.load(os.path.join(EXTENSION_PATH, "config", "extensio ...@@ -19,7 +19,7 @@ EXTENSION_TOML_DATA = toml.load(os.path.join(EXTENSION_PATH, "config", "extensio
INSTALL_REQUIRES = [ INSTALL_REQUIRES = [
# generic # generic
"numpy<2", "numpy<2",
"torch==2.4.0", "torch==2.5.1",
"onnx==1.16.1", # 1.16.2 throws access violation on Windows "onnx==1.16.1", # 1.16.2 throws access violation on Windows
"prettytable==3.3.0", "prettytable==3.3.0",
"toml", "toml",
......
...@@ -21,7 +21,7 @@ EXTENSION_TOML_DATA = toml.load(os.path.join(EXTENSION_PATH, "config", "extensio ...@@ -21,7 +21,7 @@ EXTENSION_TOML_DATA = toml.load(os.path.join(EXTENSION_PATH, "config", "extensio
INSTALL_REQUIRES = [ INSTALL_REQUIRES = [
# generic # generic
"numpy", "numpy",
"torch==2.4.0", "torch==2.5.1",
"torchvision>=0.14.1", # ensure compatibility with torch 1.13.1 "torchvision>=0.14.1", # ensure compatibility with torch 1.13.1
# 5.26.0 introduced a breaking change, so we restricted it for now. # 5.26.0 introduced a breaking change, so we restricted it for now.
# See issue https://github.com/tensorflow/tensorboard/issues/6808 for details. # See issue https://github.com/tensorflow/tensorboard/issues/6808 for details.
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment