Support torch 1.13 (#1143)

* Bug fixes for torch 1.13 * update torch_all deps --------- Co-authored-by: Benjamin Fineran <bfineran@users.noreply.github.com>
neuralmagic · Feb 14, 2023 · a69c208 · a69c208
1 parent 062715d
commit a69c208
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 6 deletions.
diff --git a/setup.py b/setup.py
@@ -62,15 +62,15 @@
 
 _onnxruntime_deps = ["onnxruntime>=1.0.0"]
 _pytorch_deps = [
-    "torch>=1.1.0,<=1.12.1",
+    "torch>=1.1.0,<=1.13.1",
     "gputils",
 ]
 _pytorch_all_deps = _pytorch_deps + [
-    "torchvision>=0.3.0,<=0.13",
-    "torchaudio<=0.12",
-    "torchvision>=0.3.0,<=0.13",
+    "torchvision>=0.3.0,<=0.14",
+    "torchaudio<=0.13",
+    "torchvision>=0.3.0,<=0.14",
 ]
-_pytorch_vision_deps = _pytorch_deps + ["torchvision>=0.3.0,<=0.13"]
+_pytorch_vision_deps = _pytorch_deps + ["torchvision>=0.3.0,<=0.14"]
 _tensorflow_v1_deps = ["tensorflow<2.0.0", "tensorboard<2.0.0", "tf2onnx>=1.0.0,<1.6"]
 _tensorflow_v1_gpu_deps = [
     "tensorflow-gpu<2.0.0",

diff --git a/src/sparseml/pytorch/base.py b/src/sparseml/pytorch/base.py
@@ -49,7 +49,7 @@
 
 
 _TORCH_MIN_VERSION = "1.0.0"
-_TORCH_MAX_VERSION = "1.12.100"  # set bug to 100 to support all future 1.9.X versions
+_TORCH_MAX_VERSION = "1.13.100"  # set bug to 100 to support all future 1.9.X versions
 
 
 def check_torch_install(