Modified dockerfile

Dockerfile

 # syntax=docker/dockerfile:1
-FROM nvidia/cuda:11.5.1-runtime-ubuntu20.04 AS builder
+FROM nvidia/cuda:11.7.1-runtime-ubuntu20.04 AS builder
 ARG PYTHON_VERSION=3.8
 SHELL ["/bin/bash", "-c"]
 # ARG SSH_KEY=docker_id_rsa
@@ -66,6 +66,8 @@ COPY . /root/catkin_ws/src/ari_hloc/
 WORKDIR /root/catkin_ws/src/ari_hloc/
 RUN mkdir -p ./third_party/Hierarchical-Localization/third_party/netvlad && mv /root/VGG16-NetVLAD-Pitts30K.mat ./third_party/Hierarchical-Localization/third_party/netvlad/
 RUN ln -s /usr/bin/python3 /usr/bin/python && bash init.sh 
+WORKDIR /root/catkin_ws/src/ari_hloc/third_party/Hierarchical-Localization
+RUN python3 setup.py install
 
 # BUILD
 WORKDIR /root/catkin_ws