xdobro23/YOLOv8-16bit
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

New train profile argument for loggers (#2862)

Browse Source 
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
This commit is contained in:
Glenn Jocher
2023-05-28 03:51:49 +02:00
committed by GitHub
parent 0bdd4ad379
commit 6391c60089
17 changed files with 76 additions and 47 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
docker/Dockerfile-arm64
View File

@ -24,7 +24,7 @@ ADD https://github.com/ultralytics/assets/releases/download/v0.0.0/yolov8n.pt /u
# Install pip packages
RUN python3 -m pip install --upgrade pip wheel
RUN pip install --no-cache -e .
RUN pip install --no-cache -e . thop
# Usage Examples -------------------------------------------------------------------------------------------------------

2
docker/Dockerfile-cpu
View File

@ -25,7 +25,7 @@ ADD https://github.com/ultralytics/assets/releases/download/v0.0.0/yolov8n.pt /u
# Install pip packages
RUN python3 -m pip install --upgrade pip wheel
RUN pip install --no-cache -e . --extra-index-url https://download.pytorch.org/whl/cpu
RUN pip install --no-cache -e . thop --extra-index-url https://download.pytorch.org/whl/cpu
# Usage Examples -------------------------------------------------------------------------------------------------------

2
docker/Dockerfile-jetson
View File

@ -25,7 +25,7 @@ ADD https://github.com/ultralytics/assets/releases/download/v0.0.0/yolov8n.pt /u
# Install pip packages manually for TensorRT compatibility https://github.com/NVIDIA/TensorRT/issues/2567
RUN python3 -m pip install --upgrade pip wheel
RUN pip install --no-cache tqdm matplotlib pyyaml psutil pandas onnx "numpy==1.23"
RUN pip install --no-cache tqdm matplotlib pyyaml psutil pandas onnx thop "numpy==1.23"
RUN pip install --no-cache -e .
# Set environment variables