mirror of https://github.com/JDAI-CV/fast-reid.git
33 lines
1.2 KiB
Docker
33 lines
1.2 KiB
Docker
# Start from the CUDA 11.3 base image
|
|
FROM nvidia/cuda:11.1.1-base-ubuntu20.04
|
|
|
|
# Install Python 3.8 and other essential packages
|
|
RUN apt-get update && apt-get install -y --no-install-recommends \
|
|
build-essential \
|
|
python3.8 \
|
|
python3-pip \
|
|
python3.8-dev \
|
|
&& apt-get clean \
|
|
&& rm -rf /var/lib/apt/lists/*
|
|
|
|
# Set the python3.8 as the default python version
|
|
RUN update-alternatives --install /usr/bin/python3 python3 /usr/bin/python3.8 1 \
|
|
&& update-alternatives --set python3 /usr/bin/python3.8
|
|
|
|
# Update pip to the latest version
|
|
RUN python3 -m pip install --upgrade pip
|
|
|
|
# Set the working directory in the container
|
|
WORKDIR /workspace/src
|
|
RUN mkdir /workspace/src/fast-reid
|
|
|
|
# Copy the requirements file into the container at /workspace/src
|
|
COPY . /workspace/src
|
|
|
|
# Install any needed packages specified in 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
|
|
|
|
# Build Cython Eval
|
|
RUN cd /workspace/src/fast-reid/fastreid/evaluation/rank_cylib && make all
|