21 lines
615 B
YAML
21 lines
615 B
YAML
services:
|
|
lhunor_torch:
|
|
image: lhunor_smpl:v2
|
|
container_name: lhunor_smpl
|
|
#command: python -c "import tensorflow as tf;tf.test.gpu_device_name()"
|
|
ipc: host
|
|
tty: true
|
|
ulimits:
|
|
memlock: -1
|
|
stack: 67108864
|
|
volumes:
|
|
- '~/workspace:/workspace'
|
|
- '/data-net/guest/lhunor/data:/data-net'
|
|
- '/data-fast/126-data1/lhunor/data:/data-local/126'
|
|
- '/data-fast/127-data1/lhunor/data:/data-local/127'
|
|
- '/data-fast/128-data1/lhunor/data:/data-local/128'
|
|
deploy:
|
|
resources:
|
|
reservations:
|
|
devices:
|
|
- capabilities: [gpu] |