Arulkumar03/Wheat_HEAD_Detection_Counting_ComputerVision_Model

1#!/bin/bash -e2# Copyright (c) Facebook, Inc. and its affiliates.3 4[[ -d "dev/packaging" ]] || {5  echo "Please run this script at detectron2 root!"6  exit 17}8 9build_one() {10  cu=$111  pytorch_ver=$212 13  case "$cu" in14    cu*)15      container_name=manylinux-cuda${cu/cu/}16      ;;17    cpu)18      container_name=manylinux-cuda10119      ;;20    *)21      echo "Unrecognized cu=$cu"22      exit 123      ;;24  esac25 26  echo "Launching container $container_name ..."27  container_id="$container_name"_"$cu"_"$pytorch_ver"28 29  py_versions=(3.7 3.8 3.9)30 31  for py in "${py_versions[@]}"; do32    docker run -itd \33      --name "$container_id" \34      --mount type=bind,source="$(pwd)",target=/detectron2 \35      pytorch/$container_name36 37    cat <<EOF | docker exec -i $container_id sh38      export CU_VERSION=$cu D2_VERSION_SUFFIX=+$cu PYTHON_VERSION=$py39      export PYTORCH_VERSION=$pytorch_ver40      cd /detectron2 && ./dev/packaging/build_wheel.sh41EOF42 43    docker container stop $container_id44    docker container rm $container_id45  done46}47 48 49if [[ -n "$1" ]] && [[ -n "$2" ]]; then50  build_one "$1" "$2"51else52  build_one cu113 1.1053  build_one cu111 1.1054  build_one cu102 1.1055  build_one cpu 1.1056 57  build_one cu111 1.958  build_one cu102 1.959  build_one cpu 1.960 61  build_one cu111 1.862  build_one cu102 1.863  build_one cu101 1.864  build_one cpu 1.865fi66