Skip to content

Commit ea90821

Browse files
committed
Install TorchVision from source
1 parent 2085998 commit ea90821

File tree

1 file changed

+18
-3
lines changed

1 file changed

+18
-3
lines changed

.github/workflows/benchmarks.yaml

Lines changed: 18 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -21,27 +21,42 @@ jobs:
2121
steps:
2222
- name: Check out repo
2323
uses: actions/checkout@v3
24+
2425
- name: Setup conda env
2526
uses: conda-incubator/setup-miniconda@v2
2627
with:
2728
auto-update-conda: true
2829
miniconda-version: "latest"
2930
activate-environment: test
3031
python-version: '3.12'
32+
3133
- name: Update pip
3234
run: python -m pip install --upgrade pip
33-
- name: Install dependencies and FFmpeg
35+
36+
- name: Install FFmpeg, PyTorch and TorchAudio
3437
run: |
3538
conda install "ffmpeg=6" pkg-config -c conda-forge
36-
python -m pip install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cpu
37-
python -m pip install decord matplotlib pandas numpy
39+
python -m pip install --pre torch torchaudio --index-url https://download.pytorch.org/whl/nightly/cpu
3840
ffmpeg -version
41+
42+
- name: Download, build and install TorchVision
43+
run: |
44+
git clone https://github.com/pytorch/vision.git
45+
cd vision
46+
python setup.py install
47+
48+
- name: Install all other dependencies
49+
run: |
50+
python -m pip install decord matplotlib pandas numpy
51+
3952
- name: Build and install torchcodec
4053
run: |
4154
python -m pip install -e ".[dev]" --no-build-isolation -vvv
55+
4256
- name: Test generic decoder benchmark
4357
run: |
4458
python benchmarks/decoders/benchmark_decoders.py --bm_video_speed_min_run_seconds 1
59+
4560
- name: TEST README data geeneration benchmark
4661
run: |
4762
python benchmarks/decoders/generate_readme_data.py --test_run

0 commit comments

Comments
 (0)