update release version and bump ci reqs

Author: speediedan

dockers/base-cuda/Dockerfile

@@ -74,18 +74,18 @@ RUN \
     # set particular PyTorch version by default
     if [[ "${CUST_BUILD}" -eq 0 ]]; then \
         CUDA_VERSION_MM=${CUDA_VERSION%.*}; \
-        uv pip install torch --prerelease=allow \
-        --find-links="https://download.pytorch.org/whl/cu${CUDA_VERSION_MM//'.'/''}/torch_stable.html" \
-        --find-links="https://download.pytorch.org/whl/test/cu${CUDA_VERSION_MM//'.'/''}/torch" \
-        --find-links="https://download.pytorch.org/whl/test/cu${CUDA_VERSION_MM//'.'/''}/pytorch-triton"; \
+        uv pip install torch --index-url https://download.pytorch.org/whl/cu${CUDA_VERSION_MM//'.'/''}; \
+        # --find-links="https://download.pytorch.org/whl/cu${CUDA_VERSION_MM//'.'/''}/torch_stable.html" \
+        # --find-links="https://download.pytorch.org/whl/test/cu${CUDA_VERSION_MM//'.'/''}/torch" \
+        # --find-links="https://download.pytorch.org/whl/test/cu${CUDA_VERSION_MM//'.'/''}/pytorch-triton"; \
     else \
         # or target a specific cuda build, by specifying a particular index url w/...
         # ... default channel
-        # uv pip install torch --torch-backend=cu128; \
+        uv pip install torch --torch-backend=cu128; \
         # ... pytorch nightly dev version
         # uv pip install --prerelease=allow torch==2.10.0.dev20251124 --index-url https://download.pytorch.org/whl/nightly/cu128; \
         # ... test channel
-        uv pip install --prerelease=allow torch==2.10.0 --index-url https://download.pytorch.org/whl/test/cu128; \
+        # uv pip install --prerelease=allow torch==2.10.0 --index-url https://download.pytorch.org/whl/test/cu128; \
     fi && \
     # We avoid installing Lightning and other dependencies here as they are usually upgraded anyway later in
     # CI but we may re-enable in the future.

dockers/docker_images_release.sh

@@ -41,7 +41,7 @@ maybe_build(){
 
 build_eval(){
 	# latest PyTorch image supported by release
-	declare -A iv=(["cuda"]="12.8.1" ["python"]="3.13" ["pytorch"]="2.10.0" ["lightning"]="2.5" ["cust_build"]="0")
+	declare -A iv=(["cuda"]="12.8.1" ["python"]="3.13" ["pytorch"]="2.10.0" ["lightning"]="2.6" ["cust_build"]="0")
 	export latest_pt="base-cu${iv["cuda"]}-py${iv["python"]}-pt${iv["pytorch"]}-pl${iv["lightning"]}"
 	export latest_azpl="py${iv["python"]}-pt${iv["pytorch"]}-pl${iv["lightning"]}-azpl-init"
 	maybe_build iv "${latest_pt}" "${latest_azpl}"

docs/source/versioning.rst

@@ -59,7 +59,7 @@ The following table shows the compatibility between Fine-Tuning Scheduler, PyTor
    * - 2.10.x
      - 2.6.0
      - 2.10.0
-     - >= 2.5.0
+     - >= 2.6.0
    * - 2.9.x
      - 2.5.0
      - 2.9.0

requirements/ci/overrides.txt

@@ -10,4 +10,4 @@
 #   uv pip install -e ".[all]" --override requirements/ci/overrides.txt --torch-backend=auto
 
 # Lightning commit pin - update this when testing against new Lightning commits
-lightning @ git+https://github.com/Lightning-AI/lightning.git@affee2885b1913ca12e4c5dd5c0da82038918c40
+lightning @ git+https://github.com/Lightning-AI/lightning.git@854037098554a5fbf6ab7c66ae9617987e82911f

requirements/ci/requirements-oldest.txt

@@ -1,5 +1,5 @@
 # This file was autogenerated by uv via the following command:
-#    uv pip compile pyproject.toml --extra all --group dev --group test --output-file /home/speediedan/repos/finetuning-scheduler/requirements/ci/requirements-oldest.txt --no-strip-extras --resolution lowest-direct --universal --python-version 3.10
+#    uv pip compile pyproject.toml --extra all --group dev --group test --output-file /home/speediedan/repos/fts-release/requirements/ci/requirements-oldest.txt --no-strip-extras --resolution lowest-direct --universal --python-version 3.10
 aiohappyeyeballs==2.6.1
     # via aiohttp
 aiohttp==3.13.3

requirements/ci/requirements.txt

@@ -1,5 +1,5 @@
 # This file was autogenerated by uv via the following command:
-#    uv pip compile pyproject.toml --extra all --group dev --group test --output-file /home/speediedan/repos/finetuning-scheduler/requirements/ci/requirements.txt --no-strip-extras --resolution highest --universal --python-version 3.10
+#    uv pip compile pyproject.toml --extra all --group dev --group test --output-file /home/speediedan/repos/fts-release/requirements/ci/requirements.txt --no-strip-extras --resolution highest --universal --python-version 3.10
 aiohappyeyeballs==2.6.1
     # via aiohttp
 aiohttp==3.13.3

scripts/build_fts_env.sh

@@ -21,6 +21,7 @@ unset target_env_name
 unset uv_install_flags
 unset no_commit_pin
 unset venv_dir
+unset torch_backend
 unset oldest
 declare -a from_source_specs=()
 
@@ -37,6 +38,7 @@ Usage: $0
    [ --uv_install_flags "flags" ]
    [ --no_commit_pin ]
    [ --venv-dir input ]
+   [ --torch-backend input ]  (cpu, cu128, auto; default: cu128 for CUDA 12.8)
    [ --from-source "package:path[:extras][:ENV_VAR=value]" ]
    [ --help ]
    Examples:
@@ -63,7 +65,7 @@ EOF
 exit 1
 }
 
-args=$(getopt -o '' --long repo_home:,target_env_name:,oldest,uv_install_flags:,no_commit_pin,venv-dir:,from-source:,help -- "$@")
+args=$(getopt -o '' --long repo_home:,target_env_name:,oldest,uv_install_flags:,no_commit_pin,venv-dir:,torch-backend:,from-source:,help -- "$@")
 if [[ $? -gt 0 ]]; then
   usage
 fi
@@ -78,6 +80,7 @@ do
     --uv_install_flags)   uv_install_flags=$2 ; shift 2 ;;
     --no_commit_pin)   no_commit_pin=1 ; shift  ;;
     --venv-dir)   venv_dir=$2 ; shift 2 ;;
+    --torch-backend)   torch_backend=$2   ; shift 2 ;;
     --from-source)
         # Accumulate multiple --from-source flags into array, join with semicolon
         from_source_specs+=("$2")
@@ -98,6 +101,9 @@ uv_install_flags=${uv_install_flags:-""}
 venv_path=$(determine_venv_path "${venv_dir}" "${target_env_name}")
 echo "Target venv path: ${venv_path}"
 
+# Set default torch backend if not specified (auto = auto-detect CUDA/CPU)
+torch_backend=${torch_backend:-"cu128"}
+
 # Join from-source specs with semicolons
 from_source_spec=""
 if [[ ${#from_source_specs[@]} -gt 0 ]]; then
@@ -189,9 +195,9 @@ fts_install(){
         # requirements.txt already has torch filtered during lock generation
         echo "Using requirements without torch (pre-installed ${TORCH_PRE_CHANNEL})"
     else
-        # Use auto torch backend for GPU detection
-        torch_backend_flag="--torch-backend=auto"
-        echo "Using torch backend: auto (GPU auto-detection)"
+        # Install stable torch - use --torch-backend for automatic backend selection
+        echo "Installing stable torch with --torch-backend=${torch_backend}..."
+        uv pip install ${uv_install_flags} torch --torch-backend=${torch_backend}
     fi
 
     uv pip install ${uv_install_flags} -e . -r "${req_file}" ${torch_backend_flag}

scripts/gen_fts_coverage.sh

@@ -15,6 +15,7 @@ unset oldest
 unset no_special
 unset run_all_and_examples
 unset allow_failures
+unset torch_backend
 declare -a from_source_specs=()
 
 SCRIPT_DIR="$(cd "$(dirname "${BASH_SOURCE[0]}")" && pwd)"
@@ -26,6 +27,7 @@ Usage: $0
    [ --repo-home input]
    [ --target-env-name input ]
    [ --oldest ]                     # Use oldest CI requirements (Python 3.10, requirements-oldest.txt)
+   [ --torch-backend input ]  (cpu, auto, cu128, etc. default: auto)
    [ --no-rebuild-base ]
    [ --no-special ]                 # Skip special tests (standalone/experimental), run only main test suite
    [ --run-all-and-examples ]       # Run all FTS example tests (both standalone and non-standalone)
@@ -56,7 +58,7 @@ EOF
 exit 1
 }
 
-args=$(getopt -o '' --long repo-home:,repo_home:,target-env-name:,target_env_name:,oldest,no-rebuild-base,no_rebuild_base,no-special,run-all-and-examples,allow-failures,include-experimental:,include_experimental,uv-install-flags:,uv_install_flags:,no-commit-pin,no_commit_pin,venv-dir:,from-source:,dry-run,help -- "$@")
+args=$(getopt -o '' --long repo-home:,repo_home:,target-env-name:,target_env_name:,oldest,torch-backend:,no-rebuild-base,no_rebuild_base,no-special,run-all-and-examples,allow-failures,include-experimental:,include_experimental,uv-install-flags:,uv_install_flags:,no-commit-pin,no_commit_pin,venv-dir:,from-source:,dry-run,help -- "$@")
 if [[ $? -gt 0 ]]; then
   usage
 fi
@@ -68,6 +70,7 @@ do
     --repo-home|--repo_home)  repo_home=$2    ; shift 2  ;;
     --target-env-name|--target_env_name)  target_env_name=$2  ; shift 2 ;;
     --oldest)   oldest=1 ; shift  ;;
+    --torch-backend)   torch_backend=$2   ; shift 2 ;;
     --no-rebuild-base|--no_rebuild_base)   no_rebuild_base=1 ; shift  ;;
     --no-special)   no_special=1 ; shift  ;;
     --run-all-and-examples)   run_all_and_examples=1 ; shift  ;;
@@ -145,6 +148,11 @@ env_rebuild(){
         cmd_args+=("--venv-dir=${venv_dir}")
     fi
 
+    # Add torch-backend flag if specified
+    if [[ -n "${torch_backend}" ]]; then
+        cmd_args+=("--torch-backend=${torch_backend}")
+    fi
+
     # Add from-source parameters
     for spec in "${from_source_specs[@]}"; do
         cmd_args+=("--from-source=${spec}")

src/finetuning_scheduler/__about__.py

@@ -1,7 +1,7 @@
 import time
 
 _this_year = time.strftime("%Y")
-__version__ = "2.10.0.dev0"
+__version__ = "2.10.0"
 __author__ = "Dan Dale"
 __author_email__ = "[email protected]"
 __license__ = "Apache-2.0"