Arulkumar03/Wheat_HEAD_Detection_Counting_ComputerVision_Model

1#!/bin/bash -e2# Copyright (c) Facebook, Inc. and its affiliates.3 4# Function to retry functions that sometimes timeout or have flaky failures5retry () {6    $*  || (sleep 1 && $*) || (sleep 2 && $*) || (sleep 4 && $*) || (sleep 8 && $*)7}8# Install with pip a bit more robustly than the default9pip_install() {10  retry pip install --progress-bar off "$@"11}12 13 14setup_cuda() {15  # Now work out the CUDA settings16  # Like other torch domain libraries, we choose common GPU architectures only.17  # See https://github.com/pytorch/pytorch/blob/master/torch/utils/cpp_extension.py18  # and https://github.com/pytorch/vision/blob/main/packaging/pkg_helpers.bash for reference.19  export FORCE_CUDA=120  case "$CU_VERSION" in21    cu113)22      export CUDA_HOME=/usr/local/cuda-11.3/23      export TORCH_CUDA_ARCH_LIST="3.7;5.0;5.2;6.0;6.1+PTX;7.0;7.5+PTX;8.0;8.6+PTX"24      ;;25    cu112)26      export CUDA_HOME=/usr/local/cuda-11.2/27      export TORCH_CUDA_ARCH_LIST="3.7;5.0;5.2;6.0;6.1+PTX;7.0;7.5+PTX;8.0;8.6+PTX"28      ;;29    cu111)30      export CUDA_HOME=/usr/local/cuda-11.1/31      export TORCH_CUDA_ARCH_LIST="3.7;5.0;5.2;6.0;6.1+PTX;7.0;7.5+PTX;8.0;8.6+PTX"32      ;;33    cu110)34      export CUDA_HOME=/usr/local/cuda-11.0/35      export TORCH_CUDA_ARCH_LIST="3.7;5.0;5.2;6.0;6.1+PTX;7.0;7.5+PTX;8.0+PTX"36      ;;37    cu102)38      export CUDA_HOME=/usr/local/cuda-10.2/39      export TORCH_CUDA_ARCH_LIST="3.7;5.0;5.2;6.0;6.1+PTX;7.0;7.5+PTX"40      ;;41    cu101)42      export CUDA_HOME=/usr/local/cuda-10.1/43      export TORCH_CUDA_ARCH_LIST="3.7;5.0;5.2;6.0;6.1+PTX;7.0;7.5+PTX"44      ;;45    cu100)46      export CUDA_HOME=/usr/local/cuda-10.0/47      export TORCH_CUDA_ARCH_LIST="3.7;5.0;5.2;6.0;6.1+PTX;7.0;7.5+PTX"48      ;;49    cu92)50      export CUDA_HOME=/usr/local/cuda-9.2/51      export TORCH_CUDA_ARCH_LIST="3.7;5.0;5.2;6.0;6.1+PTX;7.0+PTX"52      ;;53    cpu)54      unset FORCE_CUDA55      export CUDA_VISIBLE_DEVICES=56      ;;57    *)58      echo "Unrecognized CU_VERSION=$CU_VERSION"59      exit 160      ;;61  esac62}63 64setup_wheel_python() {65  case "$PYTHON_VERSION" in66    3.7) python_abi=cp37-cp37m ;;67    3.8) python_abi=cp38-cp38 ;;68    3.9) python_abi=cp39-cp39 ;;69    *)70      echo "Unrecognized PYTHON_VERSION=$PYTHON_VERSION"71      exit 172      ;;73  esac74  export PATH="/opt/python/$python_abi/bin:$PATH"75}76