@@ -74,7 +74,7 @@ case $GPU_ARCH_TYPE in
74
74
;;
75
75
esac
76
76
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} "
78
78
79
79
80
80
# 2. Install torchaudio
@@ -85,23 +85,25 @@ export BUILD_CPP_TEST=1
85
85
python setup.py install
86
86
87
87
# 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
99
101
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 .
0 commit comments