|
| 1 | +#!/bin/bash |
| 2 | + |
| 3 | +_DIR=$( cd "$( dirname "${BASH_SOURCE[0]}" )" && pwd ) |
| 4 | + |
| 5 | +# This script is for preparing the releases, conda environments, etc. |
| 6 | + |
| 7 | + |
| 8 | +CONDA_PATH="$HOME/opt/anaconda3/etc/profile.d/conda.sh" |
| 9 | + |
| 10 | + |
| 11 | +function prepare_conda_env_cpu() { |
| 12 | + source ${CONDA_PATH} |
| 13 | + conda env remove --name roosterize-cpu |
| 14 | + conda create --name roosterize-cpu python=3.7 pip -y |
| 15 | + conda activate roosterize-cpu |
| 16 | + conda install -y pytorch-cpu==1.1.0 torchvision-cpu==0.3.0 cpuonly -c pytorch |
| 17 | + pip install -r requirements.txt |
| 18 | + conda env export --no-builds > conda-envs/cpu.yml |
| 19 | +} |
| 20 | + |
| 21 | + |
| 22 | +function prepare_conda_env_gpu_cuda10() { |
| 23 | + source ${CONDA_PATH} |
| 24 | + conda env remove --name roosterize-gpu |
| 25 | + conda create --name roosterize-gpu python=3.7 pip -y |
| 26 | + conda activate roosterize-gpu |
| 27 | + conda install -y pytorch==1.1.0 torchvision==0.3.0 cudatoolkit=10.0 -c pytorch |
| 28 | + pip install -r requirements.txt |
| 29 | + conda env export --no-builds > conda-envs/gpu-cuda10.yml |
| 30 | +} |
| 31 | + |
| 32 | + |
| 33 | +function prepare_conda_env_gpu_cuda9() { |
| 34 | + source ${CONDA_PATH} |
| 35 | + conda env remove --name roosterize-gpu |
| 36 | + conda create --name roosterize-gpu python=3.7 pip -y |
| 37 | + conda activate roosterize-gpu |
| 38 | + conda install -y pytorch==1.1.0 torchvision==0.3.0 cudatoolkit=9.0 -c pytorch |
| 39 | + pip install -r requirements.txt |
| 40 | + conda env export --no-builds > conda-envs/gpu-cuda9.yml |
| 41 | +} |
| 42 | + |
| 43 | + |
| 44 | +# ========== |
| 45 | +# Main function -- program entry point |
| 46 | +# This script can be executed as ./run.sh the_function_to_run |
| 47 | + |
| 48 | +function main() { |
| 49 | + local action=${1:?Need Argument}; shift |
| 50 | + |
| 51 | + ( cd ${_DIR} |
| 52 | + $action "$@" |
| 53 | + ) |
| 54 | +} |
| 55 | + |
| 56 | +main "$@" |
0 commit comments