|
24 | 24 | conda create -n ci -y python="${PYTHON_VERSION}" |
25 | 25 | conda activate ci |
26 | 26 |
|
27 | | -# 1. Install PyTorch |
28 | | -# if [ -z "${CUDA_VERSION:-}" ] ; then |
29 | | -# if [ "${os}" == MacOSX ] ; then |
30 | | -# cudatoolkit='' |
31 | | -# else |
32 | | -# cudatoolkit="cpuonly" |
33 | | -# fi |
34 | | -# version="cpu" |
35 | | -# else |
36 | | -# version="$(python -c "print('.'.join(\"${CUDA_VERSION}\".split('.')[:2]))")" |
37 | | -# export CUDATOOLKIT_CHANNEL="nvidia" |
38 | | -# cudatoolkit="pytorch-cuda=${version}" |
39 | | -# fi |
40 | | - |
41 | | -# printf "Installing PyTorch with %s\n" "${cudatoolkit}" |
42 | | -# ( |
43 | | -# if [ "${os}" == MacOSX ] ; then |
44 | | -# # TODO: this can be removed as soon as linking issue could be resolved |
45 | | -# # see https://github.com/pytorch/pytorch/issues/62424 from details |
46 | | -# MKL_CONSTRAINT='mkl==2021.2.0' |
47 | | -# pytorch_build=pytorch |
48 | | -# else |
49 | | -# MKL_CONSTRAINT='' |
50 | | -# pytorch_build="pytorch[build="*${version}*"]" |
51 | | -# fi |
52 | | -# set -x |
53 | | - |
54 | | -# if [[ -z "$cudatoolkit" ]]; then |
55 | | -# conda install ${CONDA_CHANNEL_FLAGS:-} -y -c "pytorch-${UPLOAD_CHANNEL}" $MKL_CONSTRAINT "pytorch-${UPLOAD_CHANNEL}::${pytorch_build}" |
56 | | -# else |
57 | | -# conda install pytorch ${cudatoolkit} ${CONDA_CHANNEL_FLAGS:-} -y -c "pytorch-${UPLOAD_CHANNEL}" -c nvidia $MKL_CONSTRAINT |
58 | | -# fi |
59 | | -# ) |
60 | | - |
61 | | -export GPU_ARCH_TYPE="cpu" # TODO change this |
| 27 | +export GPU_ARCH_TYPE="cpu" |
62 | 28 |
|
63 | 29 | case $GPU_ARCH_TYPE in |
64 | 30 | cpu) |
@@ -90,14 +56,4 @@ printf "* Installing test tools\n" |
90 | 56 | conda install -y "ffmpeg<5" |
91 | 57 | python -c "import torch; import torchaudio; import torchcodec; print(torch.__version__, torchaudio.__version__, torchcodec.__version__)" |
92 | 58 |
|
93 | | -NUMBA_DEV_CHANNEL="" |
94 | | -if [[ "$(python --version)" = *3.9* || "$(python --version)" = *3.10* ]]; then |
95 | | - # Numba isn't available for Python 3.9 and 3.10 except on the numba dev channel and building from source fails |
96 | | - # See https://github.com/librosa/librosa/issues/1270#issuecomment-759065048 |
97 | | - NUMBA_DEV_CHANNEL="-c numba/label/dev" |
98 | | -fi |
99 | | -( |
100 | | - set -x |
101 | | - conda install -y -c conda-forge ${NUMBA_DEV_CHANNEL} parameterized 'requests>=2.20' |
102 | | - pip install coverage pytest pytest-cov scipy expecttest scipy "numpy>=1.26" |
103 | | -) |
| 59 | +pip3 install parameterized requests coverage pytest pytest-cov scipy numpy expecttest |
0 commit comments