|
| 1 | +#!/bin/bash |
| 2 | + |
1 | 3 | # 0. Check python and pip3
|
2 |
| -(python3 -V && pip3 -V) &>/dev/null; |
| 4 | +(python3 -V && pip3 -V) &>/dev/null |
3 | 5 | if [[ $? -ne 0 ]]; then
|
4 |
| - echo "Error: please check whether 'python3' and 'pip3' is valid"; |
5 |
| - exit 1; |
| 6 | + echo "Error: please check whether 'python3' and 'pip3' is valid" |
| 7 | + exit 1 |
6 | 8 | fi
|
7 | 9 |
|
8 | 10 | # 1. Create virtual environment
|
9 |
| -conda -V &>/dev/null; |
10 |
| -if [[ $? -eq 0 ]]; then |
11 |
| - echo "1. Creating conda environment: image_quality"; |
12 |
| - conda create --name image_quality python=3.6; |
13 |
| - source activate image_quality; |
| 11 | +conda -V &>/dev/null |
| 12 | +is_conda=$? |
| 13 | +if [[ $is_conda -eq 0 ]]; then |
| 14 | + echo "1. Creating conda environment: image_quality" |
| 15 | + conda create -y --name image_quality python=3.6 |
| 16 | + source activate image_quality |
14 | 17 | else
|
15 | 18 | echo "1. Creating virtual environment: .env"
|
16 | 19 | python3 -m venv .env
|
17 | 20 | source .env/bin/activate
|
18 | 21 | fi
|
19 | 22 |
|
20 | 23 | # 2. Install tensorflow backend
|
21 |
| -nvidia-smi &>/dev/null; |
22 |
| -if [[ $? -eq 0 ]]; then |
23 |
| - echo "2. Installing backend: tensorflow-gpu==1.12.0"; |
24 |
| - pip3 install numpy==1.15.4 Keras==2.2.4 tensorflow-gpu==1.12.0; |
| 24 | +nvidia-smi &>/dev/null |
| 25 | +is_nvidia=$? |
| 26 | +if [[ $is_nvidia -eq 0 ]]; then |
| 27 | + echo "2. Installing backend: tensorflow-gpu==1.12.0" |
| 28 | + pip install numpy==1.15.4 Keras==2.2.4 tensorflow-gpu==1.12.0 -i https://pypi.tuna.tsinghua.edu.cn/simple |
25 | 29 | else
|
26 |
| - echo "2. Installing backend: tensorflow==1.12.0"; |
27 |
| - pip3 install numpy==1.15.4 Keras==2.2.4 tensorflow==1.12.0; |
| 30 | + echo "2. Installing backend: tensorflow==1.12.0" |
| 31 | + pip install numpy==1.15.4 Keras==2.2.4 tensorflow==1.12.0 -i https://pypi.tuna.tsinghua.edu.cn/simple |
28 | 32 | fi
|
29 | 33 |
|
30 | 34 | # 3. Install dependencies
|
31 |
| -echo "3. Installing the rest dependencies"; |
32 |
| -pip3 install -r requirements.txt; |
| 35 | +echo "3. Installing the rest dependencies" |
| 36 | +pip install -r requirements.txt -i https://pypi.tuna.tsinghua.edu.cn/simple |
33 | 37 |
|
34 | 38 | # 4. Deactivate
|
35 |
| -deactivate; |
| 39 | +if [[ $is_conda -eq 0 ]]; then |
| 40 | + source deactivate |
| 41 | +else |
| 42 | + deactivate |
| 43 | +fi |
0 commit comments