Files
TensorRT/Dockerfile_full
2023-12-10 01:35:25 +08:00

26 lines
1.1 KiB
Plaintext
Executable File

# 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