2050e3a061
* [Refactor] Improve docker image build process and add some details for pre-commit * [Fix] Simple fixes from PR review
47 lines
910 B
YAML
47 lines
910 B
YAML
services:
|
|
tensorrt:
|
|
image: tensortx:1.0.0
|
|
container_name: tensortx
|
|
environment:
|
|
- NVIDIA_VISIBLE_DEVICES=all
|
|
build:
|
|
dockerfile: x86_64.dockerfile
|
|
cap_add:
|
|
- CAP_SYS_ADMIN
|
|
security_opt:
|
|
- seccomp:unconfined
|
|
privileged: true
|
|
stdin_open: true
|
|
tty: true
|
|
shm_size: '8gb'
|
|
ulimits:
|
|
memlock:
|
|
soft: -1
|
|
hard: -1
|
|
devices:
|
|
- /dev:/dev:rw
|
|
volumes:
|
|
#### user ####
|
|
- ${HOME}:/workspace/localhome:rw
|
|
#### custom ####
|
|
- mount:/mnt:rw
|
|
deploy:
|
|
restart_policy:
|
|
condition: on-failure
|
|
max_attempts: 1
|
|
delay: 5s
|
|
resources:
|
|
reservations:
|
|
devices:
|
|
- driver: nvidia
|
|
capabilities: [gpu]
|
|
count: all
|
|
|
|
volumes:
|
|
mount:
|
|
driver: local
|
|
driver_opts:
|
|
type: none
|
|
o: bind
|
|
device: ${DATA_DIR}
|