dev #7
1 changed files with 10 additions and 1 deletions
|
@ -2,7 +2,16 @@
|
|||
FROM pytorch/pytorch:2.8.0-cuda12.6-cudnn9-runtime
|
||||
|
||||
# Install system dependencies
|
||||
RUN apt update && apt install -y libgl1 && rm -rf /var/lib/apt/lists/*
|
||||
RUN apt update && apt install -y \
|
||||
libgl1 \
|
||||
libglib2.0-0 \
|
||||
libgthread-2.0-0 \
|
||||
libgstreamer1.0-0 \
|
||||
libgtk2.0-0 \
|
||||
libavcodec-dev \
|
||||
libavformat-dev \
|
||||
libswscale-dev \
|
||||
&& rm -rf /var/lib/apt/lists/*
|
||||
|
||||
# Copy and install base requirements (ML dependencies that rarely change)
|
||||
COPY requirements.base.txt .
|
||||
|
|
Loading…
Add table
Add a link
Reference in a new issue