diff --git a/.github/workflows/linux_wheel.yaml b/.github/workflows/linux_wheel.yaml index 25a5a564e..00ffa083b 100644 --- a/.github/workflows/linux_wheel.yaml +++ b/.github/workflows/linux_wheel.yaml @@ -86,7 +86,7 @@ jobs: run: python -m pip install --upgrade pip - name: Install PyTorch run: | - python -m pip install torch --index-url https://download.pytorch.org/whl/cpu + python -m pip install torch torchvision --index-url https://download.pytorch.org/whl/cpu - name: Install torchcodec from the wheel run: | wheel_path=`find pytorch/torchcodec/dist -type f -name "*.whl"` diff --git a/.github/workflows/macos_wheel.yaml b/.github/workflows/macos_wheel.yaml index 293a65124..a82bc91b7 100644 --- a/.github/workflows/macos_wheel.yaml +++ b/.github/workflows/macos_wheel.yaml @@ -87,7 +87,7 @@ jobs: - name: Install PyTorch run: | - python -m pip install torch --index-url https://download.pytorch.org/whl/cpu + python -m pip install torch torchvision --index-url https://download.pytorch.org/whl/cpu - name: Install torchcodec from the wheel run: | diff --git a/.github/workflows/windows_wheel.yaml b/.github/workflows/windows_wheel.yaml index 2fd773fec..556e08e99 100644 --- a/.github/workflows/windows_wheel.yaml +++ b/.github/workflows/windows_wheel.yaml @@ -93,7 +93,7 @@ jobs: run: python -m pip install --upgrade pip - name: Install PyTorch run: | - python -m pip install torch --index-url https://download.pytorch.org/whl/cpu + python -m pip install torch torchvision --index-url https://download.pytorch.org/whl/cpu - name: Install torchcodec from the wheel run: | wheel_path=`find pytorch/torchcodec/dist -type f -name "*.whl"`