diff --git a/Dockerfile b/Dockerfile deleted file mode 100644 index ae539cd..0000000 --- a/Dockerfile +++ /dev/null @@ -1,15 +0,0 @@ -FROM pytorch/pytorch:1.12.0-cuda11.3-cudnn8-devel - -RUN apt update && apt upgrade -y -RUN apt install git -y - -RUN git clone https://gitea.lnrweb.com/luni/reconstruction.git - -RUN conda install -c fvcore -c iopath -c conda-forge fvcore iopath -y -RUN conda install -c bottler nvidiacub -y -RUN conda install pytorch3d -c pytorch3d -y - -COPY requirements.txt /tmp/ -RUN pip install --requirement /tmp/requirements.txt - -WORKDIR /home/workspace \ No newline at end of file diff --git a/docker/Dockerfile b/docker/Dockerfile index b5317e5..52044c1 100644 --- a/docker/Dockerfile +++ b/docker/Dockerfile @@ -1,5 +1,6 @@ # Use an existing image as the base -FROM nvcr.io/nvidia/pytorch:22.06-py3 +#FROM nvcr.io/nvidia/pytorch:22.06-py3 +FROM pytorch/pytorch:1.12.0-cuda11.3-cudnn8-devel # Add the current user to the image ARG USER_NAME=lhunor diff --git a/docker/docker-compose.yml b/docker/docker-compose.yml index 19f6432..6f00263 100644 --- a/docker/docker-compose.yml +++ b/docker/docker-compose.yml @@ -11,7 +11,27 @@ services: - '/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/127-data1/lhunor/data:/data-local/127' + - '/data-fast/128-data1/lhunor/data:/data-local/128' + deploy: + resources: + reservations: + devices: + - capabilities: [gpu] + + lhunor_torch_2: + image: lhunor_reconstruction:v1.12 + container_name: lhunor_reconstruction_v1.12 + 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: diff --git a/docker/requirements.txt b/docker/requirements.txt index fa8a232..334341d 100644 --- a/docker/requirements.txt +++ b/docker/requirements.txt @@ -1,5 +1,7 @@ plotly +Cython scipy -deodr +#deodr pandas -pytorch_msssim \ No newline at end of file +pytorch_msssim +kornia \ No newline at end of file diff --git a/requirements.txt b/requirements.txt deleted file mode 100644 index fa8a232..0000000 --- a/requirements.txt +++ /dev/null @@ -1,5 +0,0 @@ -plotly -scipy -deodr -pandas -pytorch_msssim \ No newline at end of file