tyler
/
precise-lite
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Use tensorflow 2.4.4 and GPU
continuous-integration/drone/push Build was killed Details

This commit is contained in:
Tyler 2022-12-07 00:32:14 -05:00
parent 90b40394e8
commit cf59c1ffcd
2 changed files with 32 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
Dockerfile.gpu
View File

@ -1,12 +1,11 @@
FROM nvidia/cuda:10.1-cudnn7-runtime-ubuntu18.04 AS python-ubuntu
FROM nvidia/cuda:11.2.0-cudnn8-runtime-ubuntu18.04 AS python-ubuntu
RUN apt-get update
RUN apt-get install -y build-essential curl python3 python3-pip python3-venv
# Inorder to run pip3.6, run it the following way `python3.6 -m pip`
# Update Python & Install wheel
RUN python3 -m pip install pip --upgrade
RUN python3 -m pip install wheel
RUN python3 -m pip install pip --upgrade && ython3 -m pip install wheel
FROM python-ubuntu
@ -15,7 +14,7 @@ RUN apt-get update && apt-get install -y \
pkg-config portaudio19-dev swig libpulse-ocaml-dev gfortran \
libopenblas-dev libatlas-base-dev
RUN mkdir /training && pip3 install precise-lite && pip3 install tensorflow-gpu==2.3.1
RUN mkdir /training && pip3 install precise-lite && pip3 install tensorflow==2.4.4 && pip3 install tensorflow-gpu==2.4.4
WORKDIR /training

30
entrypoint.sh
View File

@ -2,4 +2,32 @@
set -e
exec $@
function populate_env() {
if [[ -z "$MODEL_PATH" ] && [ ! -z "$1" ]]; then
export MODEL_PATH="$1"
fi
if [[ -z "$DATA_FOLDER" ] && [ ! -z "$2" ]]; then
export DATA_FOLDER="$2"
fi
}
populate_env
case "$1" in
train)
precise-lite-train $TRAIN_ARGS "$MODEL_PATH" "$DATA_FOLDER"
;;
"train-incremental")
precise-lite-train-incremental $TRAIN_ARGS "$MODEL_PATH" "$DATA_FOLDER"
;;
test)
precise-lite-test "$MODEL_PATH" "$DATA_FOLDER"
;;
convert)
precise-lite-convert "$MODEL_PATH"
;;
*)
exec $@
;;
esac