This is docker file for ImageAi It's using gpu, tested on Geforce 1060, ubuntu 20.04 and Cuda 10.1. Additionaly opencv is instaled. You can find here:
- Tensorflow 1.14.0-gpu(base image)
- Keras 2.3.0
- Python3
- Opencv 4.2.0
- ImageAi 2.1.5
And more.
- Clone repository.
- Install nvidia docker support
- Go to repo directory. Type:docker build .
- To run container type:
# Runing container on gpu and mounting home to /root/tmp on docker container
docker run --gpus all -it -v /home/PC/:/root/tmp container_id bash