compiler/one-cmds/one-prepare-venv

   1 #!/bin/bash
   2
   3 # Copyright (c) 2020 Samsung Electronics Co., Ltd. All Rights Reserved
   4 #
   5 # Licensed under the Apache License, Version 2.0 (the "License");
   6 # you may not use this file except in compliance with the License.
   7 # You may obtain a copy of the License at
   8 #
   9 #    http://www.apache.org/licenses/LICENSE-2.0
  10 #
  11 # Unless required by applicable law or agreed to in writing, software
  12 # distributed under the License is distributed on an "AS IS" BASIS,
  13 # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
  14 # See the License for the specific language governing permissions and
  15 # limitations under the License.
  16
  17 set -e
  18
  19 DRIVER_PATH="$(cd "$(dirname "${BASH_SOURCE[0]}")" && pwd)"
  20
  21 VENV_ACTIVATE=${DRIVER_PATH}/venv/bin/activate
  22
  23 if [ -f ${VENV_ACTIVATE} ]; then
  24   echo "Virtual environment is already prepared."
  25   exit 0
  26 fi
  27
  28 # Install prerequisites
  29 python3 -m pip install --user -U virtualenv
  30
  31 function error_no_ensurepip ()
  32 {
  33   echo "ERROR: python3 'ensurepip' module is not found."
  34   echo "       On ubuntu, try following command:"
  35   echo
  36   echo "         apt install python$(python3 --version | awk '{print $2}' | awk -F. '{print $1"."$2}')-venv"
  37   echo
  38   echo "       You may need root privilege for this."
  39   exit 1
  40 }
  41 python3 -m ensurepip --version > /dev/null 2>&1 || error_no_ensurepip
  42
  43 # Create python virtual enviornment
  44 python3 -m venv "${DRIVER_PATH}/venv"
  45
  46 # Install tensorflow
  47 source "${VENV_ACTIVATE}"
  48
  49 python -m pip --default-timeout=1000 --trusted-host pypi.org --trusted-host files.pythonhost.org \
  50   install -U pip setuptools
  51 python -m pip --default-timeout=1000 --trusted-host pypi.org --trusted-host files.pythonhost.org \
  52   install tensorflow-cpu==2.3.0