38 lines
1.2 KiB
Docker
38 lines
1.2 KiB
Docker
# Copyright (c) 2021, NVIDIA CORPORATION & AFFILIATES. All rights reserved.
|
|
#
|
|
# NVIDIA CORPORATION and its licensors retain all intellectual property
|
|
# and proprietary rights in and to this software, related documentation
|
|
# and any modifications thereto. Any use, reproduction, disclosure or
|
|
# distribution of this software and related documentation without an express
|
|
# license agreement from NVIDIA CORPORATION is strictly prohibited.
|
|
|
|
FROM nvcr.io/nvidia/pytorch:21.08-py3
|
|
|
|
ENV PYTHONDONTWRITEBYTECODE 1
|
|
ENV PYTHONUNBUFFERED 1
|
|
|
|
RUN pip install imageio imageio-ffmpeg==0.4.4 pyspng==0.1.0
|
|
RUN pip install ipywidgets
|
|
|
|
#When X11 forwarding matplotlib
|
|
#RUN pip install cairocffi
|
|
|
|
|
|
RUN apt-get update -y
|
|
ENV TZ=Europe/Amsterdam
|
|
RUN ln -snf /usr/share/zoneinfo/$TZ /etc/localtime && echo $TZ > /etc/timezone
|
|
#RUN apt-get install -y libcairo2 python3-gi python3-gi-cairo gir1.2-gtk-3.0
|
|
RUN apt-get install -y libgirepository1.0-dev gcc libcairo2-dev pkg-config python3-dev gir1.2-gtk-3.0
|
|
RUN pip install pycairo
|
|
RUN pip install PyGObject
|
|
RUN apt-get install -y mesa-utils
|
|
|
|
# ffmpeg for cv2 video creation
|
|
RUN apt-get install -y ffmpeg
|
|
|
|
|
|
WORKDIR /workspace
|
|
|
|
RUN (printf '#!/bin/bash\nexec \"$@\"\n' >> /entry.sh) && chmod a+x /entry.sh
|
|
ENTRYPOINT ["/entry.sh"]
|