33 lines
1.1 KiB
Plaintext
33 lines
1.1 KiB
Plaintext
FROM nvcr.io/nvidia/tensorrt:24.10-py3
|
|
|
|
ARG DEBIAN_FRONTEND=noninteractive
|
|
ENV LANG=C.UTF-8
|
|
ENV LC_ALL=C.UTF-8
|
|
WORKDIR /
|
|
|
|
RUN apt-get update && apt-get install -y --no-install-recommends feh
|
|
RUN apt-get update && apt-get install -y --no-install-recommends python3-opencv
|
|
RUN pip uninstall -y opencv-python && pip install --no-cache "opencv-python<4.3"
|
|
|
|
# Show matplotlib images
|
|
RUN apt-get update && apt-get install -y --no-install-recommends python3-tk
|
|
|
|
# Update pip to allow installation of skelda in editable mode
|
|
RUN pip3 install --upgrade --no-cache-dir pip
|
|
|
|
# Install swig and later dependencies
|
|
RUN apt-get update && apt-get install -y --no-install-recommends build-essential
|
|
RUN apt-get update && apt-get install -y --no-install-recommends swig
|
|
RUN apt-get update && apt-get install -y --no-install-recommends libopencv-dev
|
|
|
|
# Install ONNX runtime
|
|
RUN pip3 install --upgrade --no-cache-dir onnxruntime-gpu
|
|
|
|
# Install skelda
|
|
RUN pip3 install --upgrade --no-cache-dir scipy
|
|
COPY ./skelda/ /skelda/
|
|
RUN pip3 install --no-cache-dir -e /skelda/
|
|
|
|
WORKDIR /RapidPoseTriangulation/
|
|
CMD ["/bin/bash"]
|