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
INSTALL_REQUIRES = [
# generic
"numpy<2",
"torch==2.4.0",
"torch==2.5.1",
"onnx==1.16.1", # 1.16.2 throws access violation on Windows
"prettytable==3.3.0",
"toml",
......
......@@ -21,7 +21,7 @@ EXTENSION_TOML_DATA = toml.load(os.path.join(EXTENSION_PATH, "config", "extensio
INSTALL_REQUIRES = [
# generic
"numpy",
"torch==2.4.0",
"torch==2.5.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.
# 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