Skip to content

Commit d69c3c6

Browse files
committed
docker: created 1 dockerfile that builds compressai-vision and vision-models studied in FCM within a cuda11.8 ubuntu22.04 with torch 2.0.0
1 parent 480f6ce commit d69c3c6

File tree

7 files changed

+50
-189
lines changed

7 files changed

+50
-189
lines changed

docker/Dockerfile

Lines changed: 50 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,50 @@
1+
#
2+
FROM nvidia/cuda:11.8.0-cudnn8-devel-ubuntu22.04
3+
4+
5+
# Remove any third-party apt sources to avoid issues with expiring keys.
6+
RUN rm -f /etc/apt/sources.list.d/*.list
7+
8+
9+
RUN apt-get update && apt-get install -y \
10+
curl \
11+
ca-certificates \
12+
sudo \
13+
git \
14+
bzip2 \
15+
libx11-6 \
16+
libpng-dev \
17+
wget \
18+
ffmpeg \
19+
python3-pip \
20+
python3-dev \
21+
cmake \
22+
build-essential \
23+
pkg-config \
24+
&& rm -rf /var/lib/apt/lists/*
25+
26+
RUN pip install uv
27+
28+
# Create a working directory.
29+
RUN mkdir /app
30+
WORKDIR /app
31+
32+
## download & compile VTM-23.3
33+
RUN mkdir /app/vtm
34+
WORKDIR /app/vtm
35+
RUN wget https://vcgit.hhi.fraunhofer.de/jvet/VVCSoftware_VTM/-/archive/VTM-23.3/VVCSoftware_VTM-VTM-23.3.tar.gz
36+
RUN tar xvf VVCSoftware_VTM-VTM-23.3.tar.gz
37+
WORKDIR /app/vtm/VVCSoftware_VTM-VTM-23.3/build
38+
RUN cmake .. -DCMAKE_BUILD_TYPE=Release
39+
RUN make -j
40+
41+
WORKDIR /app
42+
RUN git clone https://github.com/InterDigitalInc/CompressAI-Vision compressai_vision_repo && \
43+
cd compressai_vision_repo && \
44+
git submodule update --init --recursive
45+
46+
WORKDIR /app/compressai_vision_repo
47+
RUN uv python pin 3.8.20
48+
RUN uv venv
49+
RUN uv sync --only-group=dev
50+
RUN uv run -- bash scripts/install_uv.sh --no-weights

docker/vcm/Dockerfile

Lines changed: 0 additions & 74 deletions
This file was deleted.

docker/vcm/daemon.json

Lines changed: 0 additions & 8 deletions
This file was deleted.

docker/vcm/docker-driver.bash

Lines changed: 0 additions & 9 deletions
This file was deleted.

docker/vcm/make_image.bash

Lines changed: 0 additions & 5 deletions
This file was deleted.

docker/vcm/run_image.bash

Lines changed: 0 additions & 48 deletions
This file was deleted.

docker/vcm/run_image_dev.bash

Lines changed: 0 additions & 45 deletions
This file was deleted.

0 commit comments

Comments
 (0)