docker/TensorRT
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

nsight com/sys added

Browse Source
This commit is contained in:
JY Yang
2023-12-10 01:35:25 +08:00
parent df4a7b8eb1
commit c23243f42c
4 changed files with 29 additions and 9 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
Dockerfile
View File

@ -1,7 +0,0 @@
# For more detail, check NGC
FROM git.jiayudev.com:30008/docker/tensorrt:8.5.1
COPY .bashrc /home/jiayu/.bashrc

25
Dockerfile_full Executable file
View File

@ -0,0 +1,25 @@
# For more detail, check NGC
FROM git.jiayudev.com:30008/docker/tensorrt_base:8.5.1
RUN sudo sh -c 'echo kernel.perf_event_paranoid=1 > /etc/sysctl.d/local.conf'
RUN sudo apt-get update && \
sudo apt-get install -y --no-install-recommends \
libxcb-xinerama0 libxcb-icccm4 libxcb-image0 libxkbfile1 \
libxcb-keysyms1 libxcb-randr0 libxcb-render-util0 libnss3 \
libxcb-shape0 libxcb-xinput0 libxcb-cursor0 libxkbcommon-x11-0
RUN wget https://developer.download.nvidia.com/compute/cuda/repos/ubuntu2004/x86_64/cuda-keyring_1.1-1_all.deb && \
sudo dpkg -i ./cuda-keyring_1.1-1_all.deb
RUN rm ./cuda-keyring_1.1-1_all.deb
RUN sudo apt-get update
RUN sudo apt-get remove nsight-compute-2022.3.0 -y
RUN sudo apt-get update && \
sudo apt-get install -y --no-install-recommends \
cuda-nsight-compute-12-3 cuda-nsight-systems-12-3
# nsight-compute-2022.3.0 cuda-nsight-systems-11-8
RUN sudo chown jiayu:jiayu /home/jiayu/.bashrc
RUN echo "alias nsight-compute='sudo /opt/nvidia/nsight-compute/2023.3.1/ncu-ui'" >> /home/jiayu/.bashrc
RUN echo "alias nsight-system='sudo /opt/nvidia/nsight-systems/2023.3.3/bin/nsys-ui'" >> /home/jiayu/.bashrc

4
Dockerfile_origin
View File

@ -49,7 +49,7 @@ RUN rm -rf opencv_build
RUN apt-get update && apt install -y --no-install-recommends \ RUN apt-get update && apt install -y --no-install-recommends \
sudo libgflags-dev libboost-dev libxml2-dev \ sudo libgflags-dev libboost-dev libxml2-dev \
libyaml-cpp-dev sqlite3 libsqlite3-dev libboost-all-dev \ libyaml-cpp-dev sqlite3 libsqlite3-dev libboost-all-dev \
lsb-release openssh-server curl lsb-release openssh-server curl gdb bash-completion
RUN apt-get -y clean && rm -rf /var/lib/apt/lists/* RUN apt-get -y clean && rm -rf /var/lib/apt/lists/*
# install netron to show model structure # install netron to show model structure
@ -59,12 +59,14 @@ RUN pip install netron
RUN useradd -rm -c ${user} -u 1000 -d /home/${user} -s /bin/bash -G sudo ${user} RUN useradd -rm -c ${user} -u 1000 -d /home/${user} -s /bin/bash -G sudo ${user}
SHELL ["/bin/bash", "-o", "pipefail", "-c"] SHELL ["/bin/bash", "-o", "pipefail", "-c"]
RUN echo "$user ALL=(ALL) NOPASSWD:ALL" >> /etc/sudoers RUN echo "$user ALL=(ALL) NOPASSWD:ALL" >> /etc/sudoers
RUN echo '${user}:nvidia' | chpasswd
RUN apt-get update RUN apt-get update
# set working directory of trt webinar # set working directory of trt webinar
RUN mkdir -p /home/${user}/workspace/ RUN mkdir -p /home/${user}/workspace/
RUN chown -R ${user}:users /home/${user} RUN chown -R ${user}:users /home/${user}
COPY .bashrc /home/jiayu/.bashrc
# ssh setting # ssh setting
RUN systemctl enable ssh RUN systemctl enable ssh

2
scripts/start.sh
View File

@ -1 +1 @@
docker run -it --gpus '"device=0"' -it -e DISPLAY -v /dev/bus/usb:/dev/bus/usb -v /tmp/.X11-unix:/tmp/.X11-unix --net=host git.jiayudev.com:30008/docker/tensorrt:8.5.1 docker run -it --gpus '"device=0"' -it --cap-add=SYS_ADMIN -e DISPLAY -v /dev/bus/usb:/dev/bus/usb -v /tmp/.X11-unix:/tmp/.X11-unix -v /home/jiayu/2_Testing/nvidia/:/home/jiayu/nvidia --net=host git.jiayudev.com:30008/docker/tensorrt_full:8.5.1