diff --git a/docker/docker-compose.yml b/docker/docker-compose.yml new file mode 100644 index 0000000..7651492 --- /dev/null +++ b/docker/docker-compose.yml @@ -0,0 +1,24 @@ +name: "lhunor_iccv23" +services: + lhunor_tf: + image: lhunor_tf:2023.02.21 + container_name: lhunor_iccv23 + #command: python -c "import tensorflow as tf;tf.test.gpu_device_name()" + ipc: host + user: 11017:11000 + 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' + working_dir: /data-net + deploy: + resources: + reservations: + devices: + - capabilities: [gpu] \ No newline at end of file