Commit 521f1a0 1 parent 88cf777 commit 521f1a0 Copy full SHA for 521f1a0
File tree 2 files changed +14
-2
lines changed
2 files changed +14
-2
lines changed Original file line number Diff line number Diff line change @@ -100,7 +100,13 @@ echo '::endgroup::'
100
100
101
101
echo ' ::group::Install torchvision-extra-decoders'
102
102
# This can be done after torchvision was built
103
- pip install torchvision-extra-decoders --pre --index-url https://download.pytorch.org/whl/nightly/cpu
103
+ if [[ " $( uname) " == " Linux" ]]; then
104
+ extra_decoders_channel=" --pre --index-url https://download.pytorch.org/whl/nightly/cpu"
105
+ else
106
+ extra_decoders_channel=" "
107
+ fi
108
+
109
+ pip install torchvision-extra-decoders $extra_decoders_channel
104
110
echo ' ::endgroup::'
105
111
106
112
echo ' ::group::Collect environment information'
Original file line number Diff line number Diff line change 1
1
#! /bin/bash
2
2
LD_LIBRARY_PATH=" /usr/local/lib:$CUDA_HOME /lib64:$LD_LIBRARY_PATH " python packaging/wheel/relocate.py
3
3
4
- pip install torchvision-extra-decoders --pre --index-url https://download.pytorch.org/whl/nightly/cpu
4
+ if [[ " $( uname) " == " Linux" ]]; then
5
+ extra_decoders_channel=" --pre --index-url https://download.pytorch.org/whl/nightly/cpu"
6
+ else
7
+ extra_decoders_channel=" "
8
+ fi
9
+
10
+ pip install torchvision-extra-decoders $extra_decoders_channel
You can’t perform that action at this time.
0 commit comments