install.sh

# Install required packages
# Print OS information compatible with Linux, macOS and Windows
echo "Platform:" $(uname )
# Print python version
echo "Python version: $(python --version)"
# If platform has CUDA installed
if [ -x "$(command -v nvidia-smi)" ]; then
    # Print GPU name
    echo "Found GPU: $(nvidia-smi --query-gpu=name --format=csv,noheader)"
    # Print CUDA version from grepping nvidia-smi
    echo "Found CUDA version: $(nvidia-smi | grep "CUDA Version" | awk '{print $9}')"
    # Check CUDA version and format as cuXXX
    FOUND_CUDA=$(nvidia-smi | grep "CUDA Version" | awk '{print $9}' | sed 's/\.//g')
    if (( $FOUND_CUDA == 116 )); then
        CUDA_VERSION="cu116"
    elif (( $FOUND_CUDA >= 117 )); then
        CUDA_VERSION="cu117"
    else
      CUDA_VERSION="cpu"
      echo "CUDA version ${FOUND_CUDA} is not compatible. Installing PyTorch without CUDA support."
    fi
else
    echo "No GPU or CUDA installation found. Installing PyTorch without CUDA support."
    CUDA_VERSION="cpu"
fi

echo "Installing required packages."
pip install Cython numpy packaging --quiet
if [[ "$OSTYPE" == "darwin"* ]]; then
    echo "Installing PyTorch 1.13.1"
    pip install torch==1.13.1 torchvision==0.14.1 torchaudio==0.13.1 --quiet
else
    # Install pytorch
    echo "Installing PyTorch 1.13.1+${CUDA_VERSION}"
    # Install PyTorch with CUDA support
    pip install torch==1.13.1+${CUDA_VERSION} torchvision==0.14.1+${CUDA_VERSION} torchaudio==0.13.1 --extra-index-url https://download.pytorch.org/whl/${CUDA_VERSION} --quiet
fi

# Install other required packages and modules
echo "Finalizing installation ..."
pip install -e . --quiet
echo "Done !"