|
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