Skip to content

Commit

Permalink
Install TorchVision from source
Browse files Browse the repository at this point in the history
  • Loading branch information
scotts committed Nov 12, 2024
1 parent 2085998 commit ea90821
Showing 1 changed file with 18 additions and 3 deletions.
21 changes: 18 additions & 3 deletions .github/workflows/benchmarks.yaml
Original file line number Diff line number Diff line change
Expand Up @@ -21,27 +21,42 @@ jobs:
steps:
- name: Check out repo
uses: actions/checkout@v3

- name: Setup conda env
uses: conda-incubator/setup-miniconda@v2
with:
auto-update-conda: true
miniconda-version: "latest"
activate-environment: test
python-version: '3.12'

- name: Update pip
run: python -m pip install --upgrade pip
- name: Install dependencies and FFmpeg

- name: Install FFmpeg, PyTorch and TorchAudio
run: |
conda install "ffmpeg=6" pkg-config -c conda-forge
python -m pip install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cpu
python -m pip install decord matplotlib pandas numpy
python -m pip install --pre torch torchaudio --index-url https://download.pytorch.org/whl/nightly/cpu
ffmpeg -version
- name: Download, build and install TorchVision
run: |
git clone https://github.com/pytorch/vision.git
cd vision
python setup.py install
- name: Install all other dependencies
run: |
python -m pip install decord matplotlib pandas numpy
- name: Build and install torchcodec
run: |
python -m pip install -e ".[dev]" --no-build-isolation -vvv
- name: Test generic decoder benchmark
run: |
python benchmarks/decoders/benchmark_decoders.py --bm_video_speed_min_run_seconds 1
- name: TEST README data geeneration benchmark
run: |
python benchmarks/decoders/generate_readme_data.py --test_run

0 comments on commit ea90821

Please sign in to comment.