From 93139c04256e5b057b17e93377655369d80f1a51 Mon Sep 17 00:00:00 2001
From: barisx <barissenyerli@gmail.com>
Date: Sat, 12 Apr 2025 22:37:55 +0300
Subject: [PATCH] add some comments and update readme

---
 .dockerignore        |  4 ++++
 Dockerfile           | 11 +++++----
 README.md            | 53 +++++++++++++++++++++++++++++++++-----------
 scripts/test-cmds.sh |  3 ++-
 4 files changed, 53 insertions(+), 18 deletions(-)
 create mode 100644 .dockerignore

diff --git a/.dockerignore b/.dockerignore
new file mode 100644
index 0000000..8dc52ce
--- /dev/null
+++ b/.dockerignore
@@ -0,0 +1,4 @@
+install-nvidia-toolkit.sh
+docker-compose.yml
+.git*
+paper
\ No newline at end of file
diff --git a/Dockerfile b/Dockerfile
index 3958426..1e464ed 100644
--- a/Dockerfile
+++ b/Dockerfile
@@ -17,12 +17,15 @@ RUN apt upgrade --no-install-recommends -y openssl tar
 # Create working directory
 WORKDIR /app
 
-# install requirements
+# Install requirements
 COPY requirements.txt .
 RUN pip install -r requirements.txt
 
 
-# install opencv with CUDA support
-COPY scripts .
+# Install OpenCV with CUDA support
+COPY . .
 RUN rm -rf ./workspace
-RUN bash ./build_opencv.sh
\ No newline at end of file
+RUN bash scripts/build_opencv.sh
+
+# Test CUDA and OpenCV support
+RUN bash scripts/test-cmds.sh
\ No newline at end of file
diff --git a/README.md b/README.md
index f820acf..5e5fe72 100644
--- a/README.md
+++ b/README.md
@@ -36,20 +36,47 @@ MS COCO
 Docker environment (recommended)
 <details><summary> <b>Expand</b> </summary>
 
-``` shell
-# create the docker container, you can change the share memory size if you have more.
-nvidia-docker run --name yolov7 -it -v your_coco_path/:/coco/ -v your_code_path/:/yolov7 --shm-size=64g nvcr.io/nvidia/pytorch:21.08-py3
-
-# apt install required packages
-apt update
-apt install -y zip htop screen libgl1-mesa-glx
-
-# pip install required packages
-pip install seaborn thop
-
-# go to code folder
-cd /yolov7
+We create a cuda supported dockerfile for run it on docker container. First of all you need to run:
+```shell
+chmod +x install-nvidia-toolkit.sh
 ```
+Your existing machine to give a gpu capabilities to existing docker containers. It is going to download required packages after.
+```shell
+docker run --gpus all --rm -it barisx/yolov7-cuda-opencv:latest
+```
+
+After you can check it:
+```shell
+~ root# nvidia-smi
++---------------------------------------------------------------------------------------+
+| NVIDIA-SMI 535.183.01             Driver Version: 535.183.01   CUDA Version: 12.2     |
+|-----------------------------------------+----------------------+----------------------+
+| GPU  Name                 Persistence-M | Bus-Id        Disp.A | Volatile Uncorr. ECC |
+| Fan  Temp   Perf          Pwr:Usage/Cap |         Memory-Usage | GPU-Util  Compute M. |
+|                                         |                      |               MIG M. |
+|=========================================+======================+======================|
+|   0  NVIDIA GeForce RTX 2080 ...    Off | 00000000:00:00.0 Off |                  N/A |
+| N/A   43C    P0              26W /  90W |      6MiB /  8192MiB |      0%      Default |
+|                                         |                      |                  N/A |
++-----------------------------------------+----------------------+----------------------+
+                                                                                         
++---------------------------------------------------------------------------------------+
+| Processes:                                                                            |
+|  GPU   GI   CI        PID   Type   Process name                            GPU Memory |
+|        ID   ID                                                             Usage      |
+|=======================================================================================|
++---------------------------------------------------------------------------------------+
+```
+
+If you want to check test:
+```shell
+bash scripts/test-cmds.sh
+1  # -1 if it not activate
+1  # -1 if it not activate
+True  # False if it not activate
+```
+
+It will download and run gpu supported docker container that means yolov7 can reach out your existing gpu and cuda support.
 
 </details>
 
diff --git a/scripts/test-cmds.sh b/scripts/test-cmds.sh
index 0e6175c..3c2f0ad 100644
--- a/scripts/test-cmds.sh
+++ b/scripts/test-cmds.sh
@@ -1,4 +1,5 @@
 python -c "import cv2;print(cv2.cuda.getCudaEnabledDeviceCount())"
 python3 -c "import cv2;print(cv2.cuda.getCudaEnabledDeviceCount())"
 dpkg -l | grep "opencv"
-python -c "import torch;print(torch.cuda.is_available())"
\ No newline at end of file
+python -c "import torch;print(torch.cuda.is_available())"
+python -c "print("Tensor device:", target_tensor.device)"
\ No newline at end of file