forked from MCG-NJU/MixFormerV2
-
Notifications
You must be signed in to change notification settings - Fork 0
/
Dockerfile
49 lines (38 loc) · 1.26 KB
/
Dockerfile
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
# Use an Ubuntu base image with NVIDIA GPU support
FROM nvidia/cuda:11.3.1-base-ubuntu20.04
# Set environment variable to avoid interactive prompts
ARG DEBIAN_FRONTEND=noninteractive
# Install system dependencies
RUN apt-get update && apt-get install -y \
wget \
curl \
git \
sudo \
build-essential \
python3.8 \
python3.8-dev \
python3-pip \
ninja-build \
ffmpeg \
libsm6 \
libxext6 \
# Add other required packages here
&& apt-get clean \
&& rm -rf /var/lib/apt/lists/*
# Set CUDA_HOME environment variable
ENV CUDA_HOME=/usr/local/cuda
# Create a working directory
WORKDIR /mixformerv2
# Copy the install_requirements.sh script
COPY install_requirements.sh .
# Run the script to install additional requirements
RUN chmod +x install_requirements.sh && ./install_requirements.sh
# Install Python dependencies
RUN pip install --upgrade pip \
&& pip install torch==1.8.1+cu111 torchvision==0.9.1+cu111 torchaudio==0.8.1 -f https://download.pytorch.org/whl/torch_stable.html
# Copy the application code into the container
COPY ./mixformerv2 .
# Expose the desired port (if your application listens on a specific port)
EXPOSE 8002
# Set the default command to run your application
CMD ["python3.8", "./tracking/app.py"]