Skip to content

Commit 62c7fe6

Browse files
committed
Test torchcodec installation
1 parent 6c57850 commit 62c7fe6

File tree

2 files changed

+25
-23
lines changed

2 files changed

+25
-23
lines changed

.github/scripts/unittest-linux/install.sh

Lines changed: 22 additions & 20 deletions
Original file line numberDiff line numberDiff line change
@@ -74,7 +74,7 @@ case $GPU_ARCH_TYPE in
7474
;;
7575
esac
7676
PYTORCH_WHEEL_INDEX="https://download.pytorch.org/whl/${UPLOAD_CHANNEL}/${GPU_ARCH_ID}"
77-
pip install --progress-bar=off --pre torch --index-url="${PYTORCH_WHEEL_INDEX}"
77+
pip install --progress-bar=off --pre torch torchcodec --index-url="${PYTORCH_WHEEL_INDEX}"
7878

7979

8080
# 2. Install torchaudio
@@ -85,23 +85,25 @@ export BUILD_CPP_TEST=1
8585
python setup.py install
8686

8787
# 3. Install Test tools
88-
printf "* Installing test tools\n"
89-
NUMBA_DEV_CHANNEL=""
90-
if [[ "$(python --version)" = *3.9* || "$(python --version)" = *3.10* ]]; then
91-
# Numba isn't available for Python 3.9 and 3.10 except on the numba dev channel and building from source fails
92-
# See https://github.com/librosa/librosa/issues/1270#issuecomment-759065048
93-
NUMBA_DEV_CHANNEL="-c numba/label/dev"
94-
fi
95-
(
96-
set -x
97-
conda install -y -c conda-forge ${NUMBA_DEV_CHANNEL} sox libvorbis parameterized 'requests>=2.20' 'ffmpeg>=6,<7'
98-
pip install kaldi-io SoundFile librosa coverage pytest pytest-cov scipy expecttest unidecode inflect Pillow sentencepiece pytorch-lightning 'protobuf<4.21.0' demucs tinytag pyroomacoustics flashlight-text git+https://github.com/kpu/kenlm
88+
conda install -y -c conda-forge "ffmpeg=6.1.1"
89+
python -c "import torch; import torchaudio; import torchcodec; print(torch.__version__, torchaudio.__version__, torchcodec.__version__)"
90+
# printf "* Installing test tools\n"
91+
# NUMBA_DEV_CHANNEL=""
92+
# if [[ "$(python --version)" = *3.9* || "$(python --version)" = *3.10* ]]; then
93+
# # Numba isn't available for Python 3.9 and 3.10 except on the numba dev channel and building from source fails
94+
# # See https://github.com/librosa/librosa/issues/1270#issuecomment-759065048
95+
# NUMBA_DEV_CHANNEL="-c numba/label/dev"
96+
# fi
97+
# (
98+
# set -x
99+
# conda install -y -c conda-forge ${NUMBA_DEV_CHANNEL} sox libvorbis parameterized 'requests>=2.20' 'ffmpeg>=6,<7'
100+
# pip install kaldi-io SoundFile librosa coverage pytest pytest-cov scipy expecttest unidecode inflect Pillow sentencepiece pytorch-lightning 'protobuf<4.21.0' demucs tinytag pyroomacoustics flashlight-text git+https://github.com/kpu/kenlm
99101

100-
# TODO: might be better to fix the single call to `pip install` above
101-
pip install "pillow<10.0" "scipy<1.10" "numpy<2.0"
102-
)
103-
# Install fairseq
104-
git clone https://github.com/pytorch/fairseq
105-
cd fairseq
106-
git checkout e47a4c8
107-
pip install .
102+
# # TODO: might be better to fix the single call to `pip install` above
103+
# pip install "pillow<10.0" "scipy<1.10" "numpy<2.0"
104+
# )
105+
# # Install fairseq
106+
# git clone https://github.com/pytorch/fairseq
107+
# cd fairseq
108+
# git checkout e47a4c8
109+
# pip install .

.github/workflows/unittest-linux-cpu.yml

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -65,6 +65,6 @@ jobs:
6565
./.github/scripts/unittest-linux/install.sh
6666
echo '::endgroup::'
6767
68-
echo '::group::Run Tests'
69-
./.github/scripts/unittest-linux/run_test.sh
70-
echo '::endgroup::'
68+
# echo '::group::Run Tests'
69+
# ./.github/scripts/unittest-linux/run_test.sh
70+
# echo '::endgroup::'

0 commit comments

Comments
 (0)