-
By executing the following commands, it will build two Docker images:
otx:${OTX_VERSION}-cuda
andotx:${OTX_VERSION}-cuda-pretrained-ready
.git clone https://github.com/openvinotoolkit/training_extensions.git cd docker ./build.sh
-
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
.