paulasquin / docker_ubuntu_gpu_cuda_opencv

Docker image with GPU capacities with CUDA and CUDNN, along with a GPU ready OpenCV

Docker Ubuntu GPU CUDA OpenCV

Compile

First, check your GPU CUDA compute capability at this address and report the CUDA_ARCH_BIN version within Dockerfile.gpu

docker build -t ubuntu_gpu_cuda_opencv:latest -f Dockerfile.gpu .

The docker image will take a while to compile due to OpenCV GPU-ready project building. The building arguments details can be found here

Push

• First, tag the compiled image

docker tag ubuntu_gpu_cuda_opencv:latest paulasquin/ubuntu_gpu_cuda_opencv:latest

• Login

docker login

• Then push

docker push paulasquin/ubuntu_gpu_cuda_opencv:latest

Mediapipe image

If you want to compile the mediapipe ready container:

• Build with ubuntu_gpu_cuda_opencv:latest

docker build -t ubuntu_gpu_cuda_opencv:latest-mediapipe -f Dockerfile-mediapipe.gpu .

• Tag to

docker tag ubuntu_gpu_cuda_opencv:latest-mediapipe paulasquin/ubuntu_gpu_cuda_opencv:latest-mediapipe

• Then push
• Then push

docker push paulasquin/ubuntu_gpu_cuda_opencv:latest-mediapipe