[BugFix]: Limit torch version installed by ultralytics (#1580)

* [Fix] limit torch version installed by ultralytics to our max supported version

* Update max torch version to <1.14 as requested by @mgoin, this should be safe as after 1.13.1 next torch version on pypi is 2.0 https://pypi.org/project/torch/#history

* Refactor torch supported versions to a variable for a single source of truth

* Update docs with latest supported torch versions (#1593)

---------

Co-authored-by: Mark Kurtz <mark.kurtz@neuralmagic.com>
Co-authored-by: dbogunowicz <97082108+dbogunowicz@users.noreply.github.com>
Co-authored-by: Konstantin Gulin <66528950+KSGulin@users.noreply.github.com>

Author: 4 people

README.md

@@ -107,7 +107,7 @@ SparseML enables you to create a sparse model trained on your dataset in two way
 This repository is tested on Python 3.7-3.10, and Linux/Debian systems.
 
 It is recommended to install in a [virtual environment](https://docs.python.org/3/library/venv.html) to keep your system in order.
-Currently supported ML Frameworks are the following: `torch>=1.1.0,<=1.12.1`, `tensorflow>=1.8.0,<2.0.0`, `tensorflow.keras >= 2.2.0`.
+Currently supported ML Frameworks are the following: `torch>=1.1.0,<1.14`, `tensorflow>=1.8.0,<2.0.0`, `tensorflow.keras >= 2.2.0`.
 
 Install with pip using:
 

docs/source/installation.md

@@ -18,7 +18,7 @@ limitations under the License.
 
 This repository is tested on Python 3.7-3.10, and Linux/Debian systems.
 It is recommended to install in a [virtual environment](https://docs.python.org/3/library/venv.html) to keep your system in order.
-Currently supported ML Frameworks are the following: `torch>=1.1.0,<=1.8.0`, `tensorflow>=1.8.0,<=2.0.0`, `tensorflow.keras >= 2.2.0`.
+Currently supported ML Frameworks are the following: `torch>=1.1.0,<1.14`, `tensorflow>=1.8.0,<=2.0.0`, `tensorflow.keras >= 2.2.0`.
 
 Install with pip using:
 

setup.py

@@ -61,8 +61,9 @@
 _deepsparse_ent_deps = [f"deepsparse-ent~={version_nm_deps}"]
 
 _onnxruntime_deps = ["onnxruntime>=1.0.0"]
+supported_torch_version = "torch>=1.7.0,<1.14"
 _pytorch_deps = [
-    "torch>=1.1.0,<=1.13.1",
+    supported_torch_version,
     "gputils",
 ]
 _pytorch_all_deps = _pytorch_deps + [
@@ -118,7 +119,11 @@
     "tensorboardX>=1.0",
 ]
 
-_ultralytics_deps = ["ultralytics==8.0.30"]
+
+_ultralytics_deps = [
+    "ultralytics==8.0.30",
+    supported_torch_version,
+]
 
 
 def _setup_packages() -> List: