diff --git a/Dockerfile b/Dockerfile deleted file mode 100755 index d91da52..0000000 --- a/Dockerfile +++ /dev/null @@ -1,7 +0,0 @@ -# For more detail, check NGC -FROM git.jiayudev.com:30008/docker/tensorrt:8.5.1 - -COPY .bashrc /home/jiayu/.bashrc - - - diff --git a/Dockerfile_full b/Dockerfile_full new file mode 100755 index 0000000..4b42ec3 --- /dev/null +++ b/Dockerfile_full @@ -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 + + + + diff --git a/Dockerfile_origin b/Dockerfile_origin index bb397fe..88eae44 100755 --- a/Dockerfile_origin +++ b/Dockerfile_origin @@ -49,7 +49,7 @@ RUN rm -rf opencv_build RUN apt-get update && apt install -y --no-install-recommends \ sudo libgflags-dev libboost-dev libxml2-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/* # 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} SHELL ["/bin/bash", "-o", "pipefail", "-c"] RUN echo "$user ALL=(ALL) NOPASSWD:ALL" >> /etc/sudoers +RUN echo '${user}:nvidia' | chpasswd RUN apt-get update # set working directory of trt webinar RUN mkdir -p /home/${user}/workspace/ RUN chown -R ${user}:users /home/${user} +COPY .bashrc /home/jiayu/.bashrc # ssh setting RUN systemctl enable ssh diff --git a/scripts/start.sh b/scripts/start.sh index 6834847..ddd6302 100755 --- a/scripts/start.sh +++ b/scripts/start.sh @@ -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