项目作者: KeramKeram

项目描述 :
Docker file for ImageAi
高级语言: Dockerfile
项目地址: git://github.com/KeramKeram/ImageAiDocker.git
创建时间: 2020-08-09T19:36:29Z
项目社区:https://github.com/KeramKeram/ImageAiDocker

开源协议:GNU General Public License v3.0

下载


ImageAiDocker

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.

Installation

  1. Clone repository.
  2. Install nvidia docker support
  3. Go to repo directory. Type:docker build .
  4. To run container type:
  1. # Runing container on gpu and mounting home to /root/tmp on docker container
  2. docker run --gpus all -it -v /home/PC/:/root/tmp container_id bash