Skip to content
LearnIoTAI edited this page Apr 5, 2021 · 5 revisions

Container environment using Docker for learning and developing AI on the Internet of Things

  1. iotai_container_env/infer $
docker-compose build
  1. iotai_container_env/train $ Training docker image adds training and development tools to inference container.
docker-compose build
  • Run ~/.bashrc or ~/.alias
alias iotai_train='docker run --rm -u $(id -u):$(id -g) --name iotai_train --runtime=nvidia -it -v  "/etc/passwd:/etc/passwd:ro" -v  "/etc/group:/etc/group:ro" -v "/home:/home" --memory=16g  --shm-size=16g -p "780:80" -v /tmp/.X11-unix:/tmp/.X11-unix -e DISPLAY=$DISPLAY iotai_train_gpu:latest bash'
alias iotai_infer='docker run --rm -u $(id -u):$(id -g) --name iotai_infer --runtime=nvidia -it -v  "/etc/passwd:/etc/passwd:ro" -v  "/etc/group:/etc/group:ro" -v "/home:/home" --memory=16g  --shm-size=16g -p "780:80" -v /tmp/.X11-unix:/tmp/.X11-unix -e DISPLAY=$DISPLAY iotai_infer_gpu:latest bash'
Clone this wiki locally