env

Author: Borda

.lightning/workflows/fabric.yml

@@ -55,17 +55,17 @@ run: |
 
   # Verify UV installation
   command -v uv || (echo "UV not found in PATH" && exit 1)
-
-  # Alias to use system Python for all uv pip commands
-  shopt -s expand_aliases 2>/dev/null || true
-  alias uvpip="uv pip --system"
+  # Create and activate a local uv virtual environment
+  uv venv .venv -p "/usr/bin/python${python_version}" || uv venv .venv -p "python${python_version}" || uv venv .venv
+  . .venv/bin/activate
+  hash -r
 
   whereis nvidia
   nvidia-smi
   python --version
   uv --version
-  uvpip install fire wget packaging
-  uvpip list
+  uv pip install fire wget packaging
+  uv pip list
   set -ex
 
   CUDA_VERSION="${image##*cuda}" # Remove everything up to and including "cuda"
@@ -79,26 +79,26 @@ run: |
 
   if [ "${python_version}" == "3.10" ]; then
     echo "Set oldest versions"
-    uvpip uninstall -y deepspeed
-    uvpip install --upgrade "lightning-utilities[cli]"
+    uv pip uninstall -y deepspeed
+    uv pip install --upgrade "lightning-utilities[cli]"
     cd requirements/fabric
     python -m lightning_utilities.cli requirements set-oldest --req_files "['base.txt', 'strategies.txt']"
     python -m lightning_utilities.cli requirements prune-pkgs --packages deepspeed --req_files strategies.txt
     cd ../..
-    uvpip install "cython<3.0" wheel  # for compatibility
+    uv pip install "cython<3.0" wheel  # for compatibility
   fi
 
   echo "Adjust torch versions in requirements files"
   PYTORCH_VERSION=$(python -c "import torch; print(torch.__version__.split('+')[0])")
-  uvpip install wget packaging
+  uv pip install wget packaging
   python -m wget https://raw.githubusercontent.com/Lightning-AI/utilities/main/scripts/adjust-torch-versions.py
   for fpath in `ls requirements/**/*.txt`; do \
     python ./adjust-torch-versions.py $fpath ${PYTORCH_VERSION}; \
   done
 
   if [ "${PACKAGE_NAME}" == "fabric" ]; then
     echo "Replaced PL imports"
-    uvpip install --upgrade -r .actions/requirements.txt
+    uv pip install --upgrade -r .actions/requirements.txt
     python .actions/assistant.py copy_replace_imports --source_dir="./tests/tests_fabric" \
       --source_import="lightning.fabric" \
       --target_import="lightning_fabric"
@@ -108,7 +108,7 @@ run: |
   fi
 
   extra=$(python -c "print({'lightning': 'fabric-'}.get('$(PACKAGE_NAME)', ''))")
-  uvpip install -e ".[${extra}dev]" --upgrade --find-links="${TORCH_URL}"
+  uv pip install -e ".[${extra}dev]" --upgrade --find-links="${TORCH_URL}"
 
   python requirements/collect_env_details.py
   python -c "import torch ; mgpu = torch.cuda.device_count() ; assert mgpu >= 2, f'GPU: {mgpu}'"