mirrors
/
fast-reid
mirror of https://github.com/JDAI-CV/fast-reid.git
1
0
Fork 0
Code Issues Projects Releases Wiki Activity

docker and requirements updated

pull/711/head
asaid7 2023-11-15 08:22:25 -05:00
parent 537cb842ea
commit fd05fa4fc8
2 changed files with 10 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

14
docker/Dockerfile
View File

@ -18,13 +18,15 @@ RUN update-alternatives --install /usr/bin/python3 python3 /usr/bin/python3.8 1
RUN python3 -m pip install --upgrade pip RUN python3 -m pip install --upgrade pip
# Set the working directory in the container # Set the working directory in the container
WORKDIR /workspace WORKDIR /workspace/src
RUN mkdir /workspace/src/fast-reid
# Copy the requirements file into the container at /usr/src/app # Copy the requirements file into the container at /workspace/src
COPY requirements.txt ./ COPY . /workspace/src
# Install any needed packages specified in requirements.txt # Install any needed packages specified in requirements.txt
RUN python3 -m pip install --no-cache-dir -r requirements.txt RUN pip3 install torch==1.8.1+cu111 torchvision==0.9.1+cu111 torchaudio==0.8.1 -f https://download.pytorch.org/whl/torch_stable.html
RUN python3 -m pip install --no-cache-dir -r /workspace/src/fast-reid/docker/requirements.txt
# Copy the rest of your application's code into the container # Build Cython Eval
COPY . . RUN cd /workspace/src/fast-reid/fastreid/evaluation/rank_cylib && make all

1
docker/requirements.txt
View File

@ -18,3 +18,4 @@ scikit-learn
tabulate tabulate
gdown gdown
faiss-gpu faiss-gpu
Cython