Skip to content

Files

Latest commit

 

History

History

docker

Folders and files

NameName
Last commit message
Last commit date

parent directory

..
 
 
 
 
 
 
 
 

How to build cuda and cuda-pretrained-ready Docker images

  1. By executing the following commands, it will build two Docker images: otx:${OTX_VERSION}-cuda and otx:${OTX_VERSION}-cuda-pretrained-ready.

    git clone https://github.com/openvinotoolkit/training_extensions.git
    cd docker
    ./build.sh
  2. After that, you can check whether the images are built correctly such as

    docker image ls | grep otx

    Example:

    otx                                           2.0.0-cuda-pretrained-ready                    4f3b5f98f97c   3 minutes ago   14.5GB
    otx                                           2.0.0-cuda                                     8d14caccb29a   8 minutes ago   10.4GB

otx:${OTX_VERSION}-cuda is a minimal Docker image where OTX is installed with CUDA supports. On the other hand, otx:${OTX_VERSION}-cuda-pretrained-ready includes all the model pre-trained weights that OTX provides in addition to otx:${OTX_VERSION}-cuda.