24 lines
882 B
Docker
24 lines
882 B
Docker
FROM nvidia/cuda:11.0.3-cudnn8-runtime-ubuntu18.04 AS python-ubuntu
|
|
|
|
RUN apt-get update && \
|
|
apt-get install -y build-essential curl python3 python3-pip python3-venv && \
|
|
rm -rf /var/lib/apt/lists/*
|
|
|
|
# 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 liblapack-dev libpng-dev libfreetype6-dev libqhull-dev git pkg-config portaudio19-dev swig libpulse-ocaml-dev gfortran libopenblas-dev libatlas-base-dev -y && \
|
|
rm -rf /var/lib/apt/lists/*
|
|
|
|
RUN mkdir /training && pip3 install git+https://github.com/tystuyfzand/precise-lite.git@feature/tf24 && pip3 install tensorflow-gpu==2.4.4
|
|
|
|
WORKDIR /training
|
|
|
|
COPY entrypoint.sh /entrypoint.sh
|
|
|
|
ENTRYPOINT ["/entrypoint.sh"]
|
|
CMD ["/bin/bash"] |