Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Add docker support #4148

Closed
wants to merge 4 commits into from
Closed

Add docker support #4148

wants to merge 4 commits into from

Conversation

alemelis
Copy link

@alemelis alemelis commented Oct 22, 2019

I found easier to setup and train on docker rather than installing all the drivers, cuda, etc every time I switch machine (which happens too often lately)

Let me know if this is useful for you as well and I can add few more information to the readme.

The docker image (3.89 GB) is hosted here
https://hub.docker.com/r/alemelis/darknet

You can simply

./run_docker.sh -p

to get

·▄▄▄▄   ▄▄▄· ▄▄▄  ▄ •▄  ▐ ▄ ▄▄▄ .▄▄▄▄▄
██▪ ██ ▐█ ▀█ ▀▄ █·█▌▄▌▪•█▌▐█▀▄.▀·•██
▐█· ▐█▌▄█▀▀█ ▐▀▀▄ ▐▀▀▄·▐█▐▐▌▐▀▀▪▄ ▐█.▪
██. ██ ▐█ ▪▐▌▐█•█▌▐█.█▌██▐█▌▐█▄▄▌ ▐█▌·
▀▀▀▀▀•  ▀  ▀ .▀  ▀·▀  ▀▀▀ █▪ ▀▀▀  ▀▀▀

CUDA Version 10.0.130

You are running this container as user with ID 1001 and group 1001,
which should map to the ID and group for your user on the Docker host. Great!

darknet /app/darknet > 

@AlexeyAB
Copy link
Owner

@alemelis Hi, Yes, I interested in it.

Also fix this line: https://hub.docker.com/r/alemelis/darknet
RUN make OPENCV=1 GPU=1 AVX=1 OPENMP=1 CUDNN=1 OPENMP=1 -j $(nproc)
to
RUN make OPENCV=1 GPU=1 CUDNN=1 CUDNN_HALF=1 AVX=0 OPENMP=0 -j $(nproc)

@cenit
Copy link
Collaborator

cenit commented Oct 23, 2019

Could you also please share the recipe to build the docker container? That would be a very nice addition. Otherwise, it is still into my backlog, but unfortunately recently I really got too much busy

@alemelis
Copy link
Author

alemelis commented Oct 23, 2019

@AlexeyAB done
@cenit added the dockerfile to the script directory

However, ideally we'd like to do this
https://docs.docker.com/docker-hub/builds/
not sure I can link this repository to my docker-hub account, but I'm happy to try
unfortunately, it is not up to me to setup that as I'm not the admin of this repo

@cenit
Copy link
Collaborator

cenit commented Oct 23, 2019

I think that @AlexeyAB will create his own account on docker hub in order to be able to distribute containers with Darknet under his control :)

@alemelis
Copy link
Author

I think that @AlexeyAB will create his own account on docker hub in order to be able to distribute containers with Darknet under his control :)

yes, I agree 100%
feel free to close this if not needed

@cenit
Copy link
Collaborator

cenit commented Oct 23, 2019

no I think that your script is what is needed to bootstrap the activity :)

@p3jitnath
Copy link

Hi! I have recently made this repository which helps to create a Dockerfile of Darknet using OpenCV 4.2.0 (with CUDA 10.1) and cuDNN 7.6.5. Please do have a look!

This pull request was closed.
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

Successfully merging this pull request may close these issues.

4 participants