25 lines
914 B
Docker
25 lines
914 B
Docker
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 && python3 -m pip install wheel
|
|
|
|
FROM python-ubuntu
|
|
|
|
RUN apt-get update && apt-get install -y \
|
|
liblapack-dev libpng-dev libfreetype6-dev libqhull-dev git \
|
|
pkg-config portaudio19-dev swig libpulse-ocaml-dev gfortran \
|
|
libopenblas-dev libatlas-base-dev
|
|
|
|
RUN mkdir /training && pip3 install precise-lite && pip3 install tensorflow==2.4.4 && pip3 install tensorflow-gpu==2.4.4
|
|
|
|
WORKDIR /training
|
|
|
|
COPY scripts/train_incremental.py /usr/local/lib/python3.6/dist-packages/precise_lite/scripts/train_incremental.py
|
|
COPY entrypoint.sh /entrypoint.sh
|
|
|
|
ENTRYPOINT ["/entrypoint.sh"]
|
|
CMD ["/bin/bash"] |