build: Explicitly set torch minimum version as 2.6.0 (#1299)

* Explicitly set torch minimum version as 2.6.0

Signed-off-by: Charlie Truong <chtruong@nvidia.com>

* Update uv lock

Signed-off-by: chtruong814 <chtruong814@users.noreply.github.com>

---------

Signed-off-by: Charlie Truong <chtruong@nvidia.com>
Signed-off-by: chtruong814 <chtruong814@users.noreply.github.com>
Co-authored-by: chtruong814 <chtruong814@users.noreply.github.com>
Signed-off-by: NeMo Bot <nemo-bot@nvidia.com>

Author: chtruong814

docker/common/uv-pytorch.lock

@@ -3315,9 +3315,9 @@ requires-dist = [
     { name = "qwen-vl-utils", extras = ["decord"], marker = "platform_machine == 'x86_64' and sys_platform != 'darwin' and extra == 'vlm'" },
     { name = "sentencepiece", marker = "extra == 'extra'" },
     { name = "timm", marker = "extra == 'vlm'", specifier = "<=1.0.22" },
-    { name = "torch", marker = "sys_platform != 'darwin' and sys_platform != 'linux'", specifier = "<=2.10.0", index = "https://download.pytorch.org/whl/cpu" },
-    { name = "torch", marker = "sys_platform == 'darwin'", specifier = "<=2.10.0", index = "https://pypi.org/simple" },
-    { name = "torch", marker = "sys_platform == 'linux'", specifier = "<=2.10.0", index = "https://download.pytorch.org/whl/cu129" },
+    { name = "torch", marker = "sys_platform != 'darwin' and sys_platform != 'linux'", specifier = ">=2.6.0,<=2.10.0", index = "https://download.pytorch.org/whl/cpu" },
+    { name = "torch", marker = "sys_platform == 'darwin'", specifier = ">=2.6.0,<=2.10.0", index = "https://pypi.org/simple" },
+    { name = "torch", marker = "sys_platform == 'linux'", specifier = ">=2.6.0,<=2.10.0", index = "https://download.pytorch.org/whl/cu129" },
     { name = "torchao" },
     { name = "torchcodec", marker = "platform_machine == 'x86_64' and sys_platform != 'darwin' and extra == 'vlm'" },
     { name = "torchdata" },

pyproject.toml

@@ -82,7 +82,7 @@ dependencies = [
     "opencv-python-headless==4.10.0.84",
     "pybind11",
     "pyyaml",
-    "torch<=2.10.0",
+    "torch>=2.6.0,<=2.10.0",
     "torchdata",
     "transformers>=5.0.0",
     "wandb",