diff --git a/.github/workflows/linux_cuda_wheel.yaml b/.github/workflows/linux_cuda_wheel.yaml index 72d3501b..02acdccc 100644 --- a/.github/workflows/linux_cuda_wheel.yaml +++ b/.github/workflows/linux_cuda_wheel.yaml @@ -86,9 +86,6 @@ jobs: - name: Install PyTorch run: | python -m pip install --pre torch --index-url https://download.pytorch.org/whl/nightly/cu${{ env.cuda_version_without_periods }} - - name: Install cuda dependencies - run: | - conda install --yes nvidia::libnpp nvidia::cuda-nvrtc nvidia::cuda-toolkit nvidia::cuda-cudart - name: Install torchcodec from the wheel run: | wheel_path=`find pytorch/torchcodec/dist -type f -name "*.whl"` @@ -109,6 +106,9 @@ jobs: conda install "ffmpeg=${{ matrix.ffmpeg-version-for-tests }}" -c conda-forge ffmpeg -version + - name: Install cuda runtime dependencies + run: | + conda install --yes nvidia::libnpp nvidia::cuda-nvrtc nvidia::cuda-toolkit nvidia::cuda-cudart - name: Install test dependencies run: | python -m pip install --pre torchvision --index-url https://download.pytorch.org/whl/nightly/cpu