-
Notifications
You must be signed in to change notification settings - Fork 0
Home
LearnIoTAI edited this page Apr 5, 2021
·
5 revisions
-
Build
- `iotai_container_env/infer
docker-compose build
iotai_container_env/train
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'