26 lines
792 B
Docker
26 lines
792 B
Docker
FROM nvcr.io/drive/driveos-sdk/drive-agx-orin-linux-aarch64-sdk-build-x86:6.0.4.0-0004-patch-01
|
|
|
|
ENV TZ=Asia/Shanghai
|
|
ARG user=jiayu
|
|
|
|
#Force Nvidia GPU
|
|
ENV __NV_PRIME_RENDER_OFFLOAD=1
|
|
ENV __GLX_VENDOR_LIBRARY_NAME=nvidia
|
|
|
|
#Cuda TensorRT
|
|
ENV PATH="/usr/local/cuda/bin:${PATH}"
|
|
ENV LD_LIBRARY_PATH="/usr/local/cuda/lib64:${LD_LIBRARY_PATH}"
|
|
ENV LD_LIBRARY_PATH="/usr/local/tensorrt/lib:${LD_LIBRARY_PATH}"
|
|
|
|
# set userinfo
|
|
RUN useradd -rm -c ${user} -u 1000 -d /home/${user} -s /bin/bash -G sudo ${user}
|
|
SHELL ["/bin/bash", "-o", "pipefail", "-c"]
|
|
RUN echo "$user ALL=(ALL) NOPASSWD:ALL" >> /etc/sudoers
|
|
|
|
# set working directory of trt webinar
|
|
RUN mkdir -p /home/${user}/workspace/
|
|
RUN chown -R ${user}:users /home/${user}
|
|
COPY .bashrc /home/${user}/.bashrc
|
|
WORKDIR /home/${user}
|
|
USER ${user}
|