docker update

2023-05-01 19:13:14 +00:00
parent 27b70ce2be
commit 7f458fded1
2 changed files with 23 additions and 1 deletions
@@ -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]
@@ -11,3 +11,4 @@ tables
 numpy-stl
 torchmetrics
 pytorch_msssim
 importlib