diff --git a/docker/docker-compose.yml b/docker/docker-compose.yml new file mode 100644 index 0000000..d508efc --- /dev/null +++ b/docker/docker-compose.yml @@ -0,0 +1,21 @@ +name: "cvpr_2023" +services: + lhunor_torch: + image: lhunor_torch:v1.13 + container_name: lhunor_torch_v1.13 + ipc: host + tty: true + ulimits: + memlock: -1 + stack: 67108864 + volumes: + - '/home-net/lhunor:/home-net' + - '/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] diff --git a/docker/requirements.txt b/docker/requirements.txt index 977c777..eab1191 100644 --- a/docker/requirements.txt +++ b/docker/requirements.txt @@ -10,4 +10,5 @@ pytorch3d tables numpy-stl torchmetrics -pytorch_msssim \ No newline at end of file +pytorch_msssim +importlib \ No newline at end of file