build_conda.sh 1.16 KB
Newer Older
1
2
3
4
5
6
#!/bin/bash
set -ex

script_dir="$( cd "$( dirname "${BASH_SOURCE[0]}" )" >/dev/null 2>&1 && pwd )"
. "$script_dir/pkg_helpers.bash"

7
export BUILD_TYPE="conda"
8
setup_env 0.12.0
9
10
export SOURCE_ROOT_DIR="$PWD"
setup_conda_pytorch_constraint
11
setup_conda_cudatoolkit_constraint
12
setup_visual_studio_constraint
13
14

# nvidia channel included for cudatoolkit >= 11 however for 11.5 we use conda-forge
15
# HACK HACK HACK: Remove PYTHON_VERSION check once https://github.com/pytorch/builder/pull/961 is merged
16
export CUDATOOLKIT_CHANNEL="nvidia"
17
18
19
20
# NOTE: This is needed because `cudatoolkit=11.5` has a dependency on conda-forge
#       See: https://github.com/pytorch/audio/pull/2224#issuecomment-1049185550
if [[ ${CU_VERSION} = "cu115" ]]; then
    CONDA_CHANNEL_FLAGS="${CONDA_CHANNEL_FLAGS} -c conda-forge"
21
fi
22
23
24
25
26
# NOTE: There are some dependencies that are not available for macOS on Python 3.10 without conda-forge
if [[ ${OSTYPE} =~ darwin* ]] && [[ ${PYTHON_VERSION} = "3.10" ]]; then
    CONDA_CHANNEL_FLAGS="${CONDA_CHANNEL_FLAGS} -c conda-forge"
fi
conda build -c defaults -c $CUDATOOLKIT_CHANNEL ${CONDA_CHANNEL_FLAGS:-} --no-anaconda-upload --python "$PYTHON_VERSION" packaging/torchaudio