[misc] Install stable CPU-only torch wheel on CI.

duburcqa · Feb 19, 2025 · 8b7f49d · 8b7f49d
1 parent f8b9664
commit 8b7f49d
Show file tree

Hide file tree

Showing 4 changed files with 4 additions and 10 deletions.
diff --git a/.github/workflows/linux.yml b/.github/workflows/linux.yml
@@ -70,7 +70,7 @@ jobs:
         git config --global advice.detachedHead false
     - name: Install pre-compiled binaries for additional gym-jiminy dependencies
       run: |
-        "${PYTHON_EXECUTABLE}" -m pip install torch -f https://download.pytorch.org/whl/torch
+        "${PYTHON_EXECUTABLE}" -m pip install torch --no-cache-dir --index-url https://download.pytorch.org/whl/nightly/cpu
         "${PYTHON_EXECUTABLE}" -m pip install "gymnasium>=0.28,<1.1" "stable_baselines3>=2.0,<2.5"
     - name: Install latest numpy version at build-time for run-time binary compatibility
       run: |

diff --git a/.github/workflows/macos.yml b/.github/workflows/macos.yml
@@ -73,12 +73,7 @@ jobs:
         "${PYTHON_EXECUTABLE}" -m pip install delocate twine
     - name: Install pre-compiled binaries for additional gym-jiminy dependencies
       run: |
-        # FIXME: Pre-release 2.6 is needed to install torch for Python 3.13 on MacOS
-        if [[ "${{ matrix.OS }}" != 'macos-13' ]] ; then
-          "${PYTHON_EXECUTABLE}" -m pip install --pre torch --index-url https://download.pytorch.org/whl/nightly/cpu
-        else
-          "${PYTHON_EXECUTABLE}" -m pip install "torch" -f https://download.pytorch.org/whl/torch
-        fi
+        "${PYTHON_EXECUTABLE}" -m pip install torch --no-cache-dir --index-url https://download.pytorch.org/whl/nightly/cpu
         "${PYTHON_EXECUTABLE}" -m pip install "gymnasium>=0.28,<1.1" "stable_baselines3>=2.0,<2.5"
     - name: Install latest numpy version at build-time for run-time binary compatibility
       run: |

diff --git a/.github/workflows/ubuntu.yml b/.github/workflows/ubuntu.yml
@@ -75,7 +75,7 @@ jobs:
         if [[ "${{ matrix.GENERATOR }}" == 'Ninja' ]] ; then
           sudo apt install ninja-build
         fi
-        "${PYTHON_EXECUTABLE}" -m pip install torch -f https://download.pytorch.org/whl/torch
+        "${PYTHON_EXECUTABLE}" -m pip install torch --no-cache-dir --index-url https://download.pytorch.org/whl/nightly/cpu
         "${PYTHON_EXECUTABLE}" -m pip install "gymnasium>=0.28,<1.1" "stable_baselines3>=2.0,<2.5"
 
     #####################################################################################

diff --git a/.github/workflows/win.yml b/.github/workflows/win.yml
@@ -59,8 +59,7 @@ jobs:
         python -m pip install pefile machomachomangler
     - name: Install pre-compiled binaries for additional gym-jiminy dependencies
       run: |
-        # FIXME: Pre-release 2.6 is needed to install torch for Python 3.13 on Windows
-        python -m pip install --pre torch --index-url https://download.pytorch.org/whl/nightly/cpu
+        "${PYTHON_EXECUTABLE}" -m pip install torch --no-cache-dir --index-url https://download.pytorch.org/whl/nightly/cpu
         python -m pip install "gymnasium>=0.28,<1.1" "stable_baselines3>=2.0,<2.5"
     - name: Install latest numpy version at build-time for run-time binary compatibility
       run: |