update: docker

Dockerfile

@@ -1,8 +1,7 @@
 # Use an official conda-based Python image as a parent image
-FROM pytorch/pytorch:2.1.0-cuda11.8-cudnn8-runtime
-
+FROM pytorch/pytorch:2.2.1-cuda12.1-cudnn8-runtime
 LABEL maintainer vincentqyw
-ARG PYTHON_VERSION=3.10.4
+ARG PYTHON_VERSION=3.10.10
 # Set the working directory to /code
 WORKDIR /code
 # Copy the current directory contents into the container at /code
@@ -15,11 +14,7 @@ ENV PATH /opt/conda/envs/imw/bin:$PATH
 # Make RUN commands use the new environment
 SHELL ["conda", "run", "-n", "imw", "/bin/bash", "-c"]
 RUN pip install --upgrade pip
-
-#RUN pip install torch==2.0.1 torchvision==0.15.2 torchaudio==2.0.2 --index-url https://download.pytorch.org/whl/cu118
-RUN pip install torch==2.1.0 torchvision==0.16.0 --index-url https://download.pytorch.org/whl/cu118
 RUN pip install -r env-docker.txt
-
 RUN apt-get update && apt-get install ffmpeg libsm6 libxext6 -y
 
 # Export port

build_docker.sh

@@ -1,4 +1,3 @@
 docker build -t image-matching-webui:latest . --no-cache
 docker tag image-matching-webui:latest vincentqin/image-matching-webui:latest
-docker push vincentqin/image-matching-webui:latest
- 
+docker push vincentqin/image-matching-webui:latest

env-docker.txt

@@ -27,5 +27,6 @@ seaborn==0.12.2
 shapely==2.0.1
 tensorboardX==2.6.1
 torchmetrics==0.6.0
+torchvision==0.17.1
 tqdm==4.65.0
-yacs==0.1.8
+yacs==0.1.8