build_conda.sh 607 Bytes
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
9
10
11
export SOURCE_ROOT_DIR="$PWD"
setup_conda_pytorch_constraint
setup_conda_cudatoolkit_constraint
12
setup_visual_studio_constraint
13
setup_junit_results_folder
14
15
export CUDATOOLKIT_CHANNEL="nvidia"

16
17
18
19
20
if [[ "$PYTHON_VERSION" == "3.11" ]]; then
  export CONDA_CHANNEL_FLAGS="${CONDA_CHANNEL_FLAGS} -c malfet"
fi

conda build -c $CUDATOOLKIT_CHANNEL $CONDA_CHANNEL_FLAGS --no-anaconda-upload --no-test --python "$PYTHON_VERSION" packaging/torchvision