hienanh/pysot_python repository overview

docker build . -t hienanh/pysot_python:py3.10-torch2.5.0-cuda12.4-cudnn9-runtime
docker_nvidia --name pysot_python_dev -v $(pwd):/workspace/pysot hienanh/pysot_python:py3.10-torch2.5.0-cuda12.4-cudnn9-runtime

FROM pytorch/pytorch:2.5.0-cuda12.4-cudnn9-runtime
ENV DEBIAN_FRONTEND=noninteractive

# Update packages and install required system dependencies
RUN apt update && apt install -y libgl1-mesa-glx libglib2.0-0 \
    libxkbcommon-x11-0 qtbase5-dev qtchooser qt5-qmake qtbase5-dev-tools \
    git gcc wget python3-dev

# Install Python 3.10 and set it in the base environment
RUN conda install python=3.10 -n base -y

# Copy requirements and install Python dependencies
COPY requirements.txt requirements.txt
RUN conda install -n base ipykernel --update-deps --force-reinstall
RUN python3 -m pip install torch==2.5.0+cu124 torchvision==0.20.0+cu124 torchaudio==2.5.0+cu124 --extra-index-url https://download.pytorch.org/whl/cu124 && \
    python3 -m pip install -r requirements.txt

# Set working directory
WORKDIR /workspace/pysot
CMD rm -rf toolkit/utils/region.c toolkit/utils/region.cpython-3* && python3 setup.py build_ext --inplace && bash