Update dependencies and add new package indices

https://docs.astral.sh/uv/guides/integration/pytorch/#installing-pytorch

Author: micedre

pyproject.toml

@@ -21,7 +21,6 @@ dependencies = [
     "torch>=2.11.0",
     "torchvision>=0.26.0",
     "transformers>=5.5.1",
-    "nvidia-cuda-runtime==12.0.90",
 ]
 
 authors = [
@@ -46,6 +45,12 @@ gdal = [
   { index = "gdal-wheels", marker = "sys_platform == 'linux'" },
   { index = "geospatial_wheels", marker = "sys_platform == 'win32'" },
 ]
+torch = [
+  { index = "pytorch-cu128", marker = "sys_platform == 'linux' or sys_platform == 'win32'" },
+]
+torchvision = [
+  { index = "pytorch-cu128", marker = "sys_platform == 'linux' or sys_platform == 'win32'" },
+]
 
 [[tool.uv.index]]
 name = "geospatial_wheels"
@@ -56,3 +61,8 @@ explicit = true
 name = "gdal-wheels"
 url = "https://gitlab.com/api/v4/projects/61637378/packages/pypi/simple"
 explicit = true
+
+[[tool.uv.index]]
+name = "pytorch-cu128"
+url = "https://download.pytorch.org/whl/cu128"
+explicit = true