From 5c30bcf29086b4bb71ac355725509af1fc34ddf7 Mon Sep 17 00:00:00 2001 From: Ar-Ray-code Date: Tue, 20 Sep 2022 14:53:53 +0900 Subject: [PATCH 1/3] update Dockerfile --- Dockerfile | 7 ++++--- README.md | 30 +++++++++++++++--------------- 2 files changed, 19 insertions(+), 18 deletions(-) diff --git a/Dockerfile b/Dockerfile index 0e91986..c8ffe93 100644 --- a/Dockerfile +++ b/Dockerfile @@ -1,4 +1,4 @@ -FROM nvidia/cuda:11.7.0-devel-ubuntu22.04 +FROM nvidia/cuda:11.7.0-cudnn8-devel-ubuntu22.04 ENV TZ=Asia/Tokyo RUN ln -snf /usr/share/zoneinfo/$TZ /etc/localtime && echo $TZ > /etc/timezone @@ -23,6 +23,7 @@ RUN apt update && \ apt install -y git && \ apt install -y xterm && \ apt install -y wget && \ + apt install -y pciutils && \ apt -y clean && \ rm -rf /var/lib/apt/lists/* @@ -33,11 +34,11 @@ RUN bash darknet_ros/rm_darknet_CMakeLists.sh WORKDIR /home/ros2_ws/ COPY ./yolov4-tiny-docker.bash /home/ros2_ws/yolov4-tiny-docker.bash -RUN ln -s /usr/local/cuda/lib64/stubs/libcuda.so /usr/local/cuda/lib64/stubs/libcuda.so.1 +# RUN ln -s /usr/local/cuda/lib64/stubs/libcuda.so /usr/local/cuda/lib64/stubs/libcuda.so.1 RUN LD_LIBRARY_PATH=/usr/local/cuda/lib64/stubs/:$LD_LIBRARY_PATH # USE Usb Camera CMD ["bash"] # ================================ # docker build . -t darknet-ros-fp16 -# docker run --rm -it --device /dev/video0:/dev/video0:mwr -e DISPLAY=$DISPLAY --runtime nvidia -v /tmp/.X11-unix:/tmp/.X11-unix darknet-ros-fp16 /bin/bash \ No newline at end of file +# docker run --rm -it --device /dev/video0:/dev/video0:mwr -e DISPLAY=$DISPLAY --gps all -v /tmp/.X11-unix:/tmp/.X11-unix darknet-ros-fp16 /bin/bash \ No newline at end of file diff --git a/README.md b/README.md index 1e88e1d..c7a148c 100644 --- a/README.md +++ b/README.md @@ -43,21 +43,6 @@ darknet_ros + ROS2 Humble + OpenCV4 + CUDA 11 + __CUDNN (FP16)__ :fire::fire::fi
- - - ## Installation 🐢 @@ -73,6 +58,21 @@ $ darknet_ros_yolov4/darknet_ros/rm_darknet_CMakeLists.sh $ cd ~/ros2_ws $ colcon build --symlink-install ``` + +### NVIDIA-Docker + +- Driver version : 515.65.01 +- NVIDIA Docker2 +- NVIDIA Graphics card (Tested : RTX3060Ti) + +```bash +git clone https://github.com/Ar-Ray-code/darknet_ros_yolov4.git +docker build -t darknet-ros-fp16 ./darknet_ros_fp16/. + +# connect webcamera +docker run --rm -it --device /dev/video0:/dev/video0:mwr -e DISPLAY=$DISPLAY --gps all -v /tmp/.X11-unix:/tmp/.X11-unix darknet-ros-fp16 /bin/bash +``` + ### Edit CMakeLists.txt #### Options From 5e43bdb897857cc32414e0a180fabb59b92ad27f Mon Sep 17 00:00:00 2001 From: Ar-Ray-code Date: Tue, 20 Sep 2022 14:55:55 +0900 Subject: [PATCH 2/3] fix typo --- Dockerfile | 6 ++++-- README.md | 2 +- 2 files changed, 5 insertions(+), 3 deletions(-) diff --git a/Dockerfile b/Dockerfile index c8ffe93..6a8d04d 100644 --- a/Dockerfile +++ b/Dockerfile @@ -40,5 +40,7 @@ RUN LD_LIBRARY_PATH=/usr/local/cuda/lib64/stubs/:$LD_LIBRARY_PATH # USE Usb Camera CMD ["bash"] # ================================ -# docker build . -t darknet-ros-fp16 -# docker run --rm -it --device /dev/video0:/dev/video0:mwr -e DISPLAY=$DISPLAY --gps all -v /tmp/.X11-unix:/tmp/.X11-unix darknet-ros-fp16 /bin/bash \ No newline at end of file +# git clone https://github.com/Ar-Ray-code/darknet_ros_yolov4.git +# docker build -t darknet-ros-fp16 ./darknet_ros_fp16/. +# # (connect webcamera) +# docker run --rm -it --device /dev/video0:/dev/video0:mwr -e DISPLAY=$DISPLAY --gpus all -v /tmp/.X11-unix:/tmp/.X11-unix darknet_ros_fp16 /bin/bash \ No newline at end of file diff --git a/README.md b/README.md index c7a148c..692d63a 100644 --- a/README.md +++ b/README.md @@ -70,7 +70,7 @@ git clone https://github.com/Ar-Ray-code/darknet_ros_yolov4.git docker build -t darknet-ros-fp16 ./darknet_ros_fp16/. # connect webcamera -docker run --rm -it --device /dev/video0:/dev/video0:mwr -e DISPLAY=$DISPLAY --gps all -v /tmp/.X11-unix:/tmp/.X11-unix darknet-ros-fp16 /bin/bash +docker run --rm -it --device /dev/video0:/dev/video0:mwr -e DISPLAY=$DISPLAY --gpus all -v /tmp/.X11-unix:/tmp/.X11-unix darknet_ros_fp16 /bin/bash ``` ### Edit CMakeLists.txt From dfe03543e91b5d0448c27e57c84d2141dbdea89c Mon Sep 17 00:00:00 2001 From: Ar-Ray-code Date: Tue, 20 Sep 2022 14:57:58 +0900 Subject: [PATCH 3/3] update readme --- Dockerfile | 4 ++-- README.md | 4 ++-- 2 files changed, 4 insertions(+), 4 deletions(-) diff --git a/Dockerfile b/Dockerfile index 6a8d04d..9bc1d67 100644 --- a/Dockerfile +++ b/Dockerfile @@ -40,7 +40,7 @@ RUN LD_LIBRARY_PATH=/usr/local/cuda/lib64/stubs/:$LD_LIBRARY_PATH # USE Usb Camera CMD ["bash"] # ================================ -# git clone https://github.com/Ar-Ray-code/darknet_ros_yolov4.git -# docker build -t darknet-ros-fp16 ./darknet_ros_fp16/. +# git clone https://github.com/Ar-Ray-code/darknet_ros_fp16.git +# docker build -t darknet_ros_fp16 ./darknet_ros_fp16/. # # (connect webcamera) # docker run --rm -it --device /dev/video0:/dev/video0:mwr -e DISPLAY=$DISPLAY --gpus all -v /tmp/.X11-unix:/tmp/.X11-unix darknet_ros_fp16 /bin/bash \ No newline at end of file diff --git a/README.md b/README.md index 692d63a..820ce98 100644 --- a/README.md +++ b/README.md @@ -66,8 +66,8 @@ $ colcon build --symlink-install - NVIDIA Graphics card (Tested : RTX3060Ti) ```bash -git clone https://github.com/Ar-Ray-code/darknet_ros_yolov4.git -docker build -t darknet-ros-fp16 ./darknet_ros_fp16/. +git clone https://github.com/Ar-Ray-code/darknet_ros_fp16.git +docker build -t darknet_ros_fp16 ./darknet_ros_fp16/. # connect webcamera docker run --rm -it --device /dev/video0:/dev/video0:mwr -e DISPLAY=$DISPLAY --gpus all -v /tmp/.X11-unix:/tmp/.X11-unix darknet_ros_fp16 /bin/bash