.

pytorch · Oct 18, 2024 · 1c0b022 · 1c0b022
1 parent 83b34b4
commit 1c0b022
Showing 1 changed file with 8 additions and 8 deletions.
diff --git a/.github/workflows/linux_cuda_wheel.yaml b/.github/workflows/linux_cuda_wheel.yaml
@@ -84,15 +84,15 @@ jobs:
           activate-environment: test
           python-version: ${{ matrix.python-version }}
       - name: Update pip
-        run: python -m pip install --upgrade pip
+        run: ${CONDA_RUN} python -m pip install --upgrade pip
       - name: Install PyTorch
         run: |
-          python -m pip install --pre torch --index-url https://download.pytorch.org/whl/nightly/cu${{ env.cuda_version_without_periods }}
+          ${CONDA_RUN} python -m pip install --pre torch --index-url https://download.pytorch.org/whl/nightly/cu${{ env.cuda_version_without_periods }}
       - name: Install torchcodec from the wheel
         run: |
           wheel_path=`find pytorch/torchcodec/dist -type f -name "*.whl"`
           echo Installing $wheel_path
-          python -m pip install $wheel_path -vvv
+          ${CONDA_RUN} python -m pip install $wheel_path -vvv
 
       - name: Check out repo
         uses: actions/checkout@v3
@@ -105,17 +105,17 @@ jobs:
           source packaging/helpers.sh
           assert_ffmpeg_not_installed
 
-          conda install "ffmpeg=${{ matrix.ffmpeg-version-for-tests }}" -c conda-forge
-          ffmpeg -version
+          ${CONDA_RUN} conda install "ffmpeg=${{ matrix.ffmpeg-version-for-tests }}" -c conda-forge
+          ${CONDA_RUN} ffmpeg -version
 
       - name: Install cuda runtime dependencies
         run: |
-          conda install --yes nvidia::libnpp nvidia::cuda-nvrtc=12.4 nvidia::cuda-toolkit=12.4 nvidia::cuda-cudart=12.4 nvidia::cuda-driver-dev=12.4
+          ${CONDA_RUN} conda install --yes nvidia::libnpp nvidia::cuda-nvrtc=12.4 nvidia::cuda-toolkit=12.4 nvidia::cuda-cudart=12.4 nvidia::cuda-driver-dev=12.4
       - name: Install test dependencies
         run: |
-          python -m pip install --pre torchvision --index-url https://download.pytorch.org/whl/nightly/cpu
+          ${CONDA_RUN} python -m pip install --pre torchvision --index-url https://download.pytorch.org/whl/nightly/cpu
           # Ideally we would find a way to get those dependencies from pyproject.toml
-          python -m pip install numpy pytest pillow
+          ${CONDA_RUN} python -m pip install numpy pytest pillow
 
       - name: Delete the src/ folder just for fun
         run: |