OrgLion-ML
/
darknet
mirror of https://github.com/AlexeyAB/darknet.git
1
0
Fork 0
Code Issues Projects Releases Wiki Activity

Fixes #8897 (#8901)

This commit is contained in:
Stefano Sinigardi 2024-05-03 15:41:13 +02:00 committed by GitHub
parent ffdd5e41a8
commit 5dc0126e2f
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
3 changed files with 9 additions and 11 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
Dockerfile.cpu
View File

@ -14,7 +14,7 @@ RUN rm Dockerfile.cpu
RUN rm Dockerfile.gpu
RUN rm Docker-compose.yml
RUN rm docker-compose.yml
RUN make
@ -28,7 +28,7 @@ RUN apt-get install -y sudo libgomp1
RUN useradd -U -m yolo
RUN usermod -aG sudo yolo
RUN usermod -aG sudo yolo
RUN usermod --shell /bin/bash yolo
@ -45,5 +45,3 @@ RUN ldconfig
WORKDIR /home/yolo/darknet
USER yolo

8
Dockerfile.gpu
View File

@ -14,11 +14,11 @@ RUN rm Dockerfile.cpu
RUN rm Dockerfile.gpu
RUN rm Docker-compose.yml
RUN rm docker-compose.yml
RUN make
FROM nvidia/cuda:11.6.0-cudnn8-devel-ubuntu20.04
FROM nvidia/cuda:11.6.1-cudnn8-devel-ubuntu20.04
ENV DEBIAN_FRONTEND noninteractive
@ -28,7 +28,7 @@ RUN apt-get install -y sudo libgomp1
RUN useradd -U -m yolo
RUN usermod -aG sudo yolo
RUN usermod -aG sudo yolo
RUN usermod --shell /bin/bash yolo
@ -44,4 +44,4 @@ RUN ldconfig
WORKDIR /home/yolo/darknet
USER yolo
USER yolo

6
docker-compose.yml
View File

@ -2,7 +2,7 @@ version: '2'
services:
yolo-gpu:
build:
build:
context: .
dockerfile: Dockerfile.gpu
image: yolo:gpu
@ -14,7 +14,7 @@ services:
count: 1
capabilities: [gpu]
yolo-cpu:
build:
build:
context: .
dockerfile: Dockerfile.cpu
image: yolo:cpu
image: yolo:cpu