Skip to content

Install torchvision-extra-decoders from nightly channel #8963

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Merged
merged 10 commits into from
Mar 17, 2025
Merged

Install torchvision-extra-decoders from nightly channel #8963

Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
10 commits
Select commit Hold shift + click to select a range
88cf777
Install torchvision-extra-decoders from nightly channel
NicolasHug Mar 12, 2025
521f1a0
Only set channel on linux
NicolasHug Mar 12, 2025
b288574
Merge branch 'main' into extra-nightly
NicolasHug Mar 17, 2025
fbb843c
Merge branch 'main' into extra-nightly
NicolasHug Mar 17, 2025
b38ef2f
Merge branch 'main' of github.com:pytorch/vision into extra-nightly
NicolasHug Mar 17, 2025
316bf4b
Merge branch 'main' of github.com:pytorch/vision into extra-nightly
NicolasHug Mar 17, 2025
517d0b3
Dont install on aarch64
NicolasHug Mar 17, 2025
caa69ed
Merge branch 'extra-nightly' of github.com:nicolashug/vision into ext…
NicolasHug Mar 17, 2025
3a2e90b
Try again
NicolasHug Mar 17, 2025
726eec8
Merge branch 'main' into extra-nightly
NicolasHug Mar 17, 2025
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
8 changes: 7 additions & 1 deletion .github/scripts/setup-env.sh
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -100,7 +100,13 @@ echo '::endgroup::'

echo '::group::Install torchvision-extra-decoders'
# This can be done after torchvision was built
pip install torchvision-extra-decoders
if [[ "$(uname)" == "Linux" && "$(uname -m)" != "aarch64" ]]; then
extra_decoders_channel="--pre --index-url https://download.pytorch.org/whl/nightly/cpu"
else
extra_decoders_channel=""
fi

pip install torchvision-extra-decoders $extra_decoders_channel
echo '::endgroup::'

echo '::group::Collect environment information'
Expand Down
8 changes: 7 additions & 1 deletion packaging/post_build_script.sh
View file
Open in desktop
Original file line number Diff line number Diff line change
@@ -1,4 +1,10 @@
#!/bin/bash
LD_LIBRARY_PATH="/usr/local/lib:$CUDA_HOME/lib64:$LD_LIBRARY_PATH" python packaging/wheel/relocate.py

pip install torchvision-extra-decoders
if [[ "$(uname)" == "Linux" && "$(uname -m)" != "aarch64" ]]; then
extra_decoders_channel="--pre --index-url https://download.pytorch.org/whl/nightly/cpu"
else
extra_decoders_channel=""
fi

pip install torchvision-extra-decoders $extra_decoders_channel
Loading