Skip to content

chore: unify projects' Docker files #184

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Merged
amadeuszsz merged 2 commits into from
Feb 13, 2026
+23 −22
Merged

chore: unify projects' Docker files #184

Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
2 commits
Select commit Hold shift + click to select a range
d34e3b9
chore: unify projects' Docker files
amadeuszsz Feb 9, 2026
2bd5e3d
refactor: follow image & container naming convention
amadeuszsz Feb 9, 2026
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
2 changes: 1 addition & 1 deletion docs/tutorial/tutorial_detection_3d.md
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -64,5 +64,5 @@ python projects/CenterPoint/scripts/deploy.py projects/CenterPoint/configs/t4dat
- You may need `sudo` to use `docker` command.

```sh
DOCKER_BUILDKIT=1 docker build -t autoware-ml .
DOCKER_BUILDKIT=1 docker build -t awml .
```
2 changes: 1 addition & 1 deletion projects/CalibrationStatusClassification/Dockerfile
View file
Open in desktop
Original file line number Diff line number Diff line change
@@ -1,4 +1,4 @@
ARG AWML_BASE_IMAGE="autoware-ml:latest"
ARG AWML_BASE_IMAGE="ghcr.io/tier4/awml:base-latest"
FROM ${AWML_BASE_IMAGE}
ARG TRT_VERSION=10.8.0.43

Expand Down
4 changes: 2 additions & 2 deletions projects/CalibrationStatusClassification/README.md
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -25,13 +25,13 @@
- Docker build for CalibrationStatusClassification

```sh
DOCKER_BUILDKIT=1 docker build -t autoware-ml-calib projects/CalibrationStatusClassification/
DOCKER_BUILDKIT=1 docker build -t awml-calib projects/CalibrationStatusClassification/
```

- Run docker

```sh
docker run -it --rm --gpus all --shm-size=64g --name awml-calib -p 6006:6006 -v $PWD/:/workspace -v $PWD/data:/workspace/data autoware-ml-calib
docker run -it --rm --gpus all --shm-size=64g --name awml-calib -p 6006:6006 -v $PWD/:/workspace -v $PWD/data:/workspace/data awml-calib
```

### 2. Config
Expand Down
9 changes: 4 additions & 5 deletions projects/FRNet/Dockerfile
View file
Open in desktop
Original file line number Diff line number Diff line change
@@ -1,6 +1,6 @@
FROM autoware-ml

ARG TRT_MAJOR=10.4.0
ARG AWML_BASE_IMAGE="ghcr.io/tier4/awml:base-latest"
FROM ${AWML_BASE_IMAGE}
ARG TRT_VERSION=10.8.0.43

# Install pip dependencies
RUN python3 -m pip --no-cache-dir install \
Expand All @@ -9,8 +9,7 @@ RUN python3 -m pip --no-cache-dir install \
onnx_graphsurgeon \
polygraphy \
pycuda \
tensorrt==${TRT_MAJOR} \
tensorrt-cu12==${TRT_MAJOR}
tensorrt-cu12==${TRT_VERSION}

WORKDIR /workspace
RUN pip install --no-cache-dir -e .
4 changes: 2 additions & 2 deletions projects/FRNet/README.md
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -27,13 +27,13 @@
- Docker build for FRNet

```sh
DOCKER_BUILDKIT=1 docker build -t autoware-ml-frnet projects/FRNet/
DOCKER_BUILDKIT=1 docker build -t awml-frnet projects/FRNet/
```

- Run docker

```sh
docker run -it --rm --gpus all --shm-size=64g --name awml -p 6006:6006 -v $PWD/:/workspace -v $PWD/data:/workspace/data autoware-ml-frnet
docker run -it --rm --gpus all --shm-size=64g --name awml -p 6006:6006 -v $PWD/:/workspace -v $PWD/data:/workspace/data awml-frnet
```

### 2. Config
Expand Down
2 changes: 1 addition & 1 deletion projects/PTv3/Dockerfile
View file
Open in desktop
Original file line number Diff line number Diff line change
@@ -1,4 +1,4 @@
ARG AWML_BASE_IMAGE="autoware-ml"
ARG AWML_BASE_IMAGE="ghcr.io/tier4/awml:base-latest"
FROM ${AWML_BASE_IMAGE}

ENV FLASH_ATTN_CUDA_ARCHS="120"
Expand Down
4 changes: 2 additions & 2 deletions projects/PTv3/README.md
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -23,13 +23,13 @@ AWML's implementation is a port of the [original code](https://github.com/Pointc
- This project requires a different docker environment that most other projects.

```sh
DOCKER_BUILDKIT=1 docker build -t autoware-ml-ptv3 -f projects/PTv3/Dockerfile . --progress=plain
DOCKER_BUILDKIT=1 docker build -t awml-ptv3 -f projects/PTv3/Dockerfile . --progress=plain
```

-Run docker

```sh
docker run -it --rm --gpus all --shm-size=64g --name awml-ptv3 -p 6006:6006 -v $PWD/:/workspace -v $PWD/data:/workspace/data autoware-ml-ptv3
docker run -it --rm --gpus all --shm-size=64g --name awml-ptv3 -p 6006:6006 -v $PWD/:/workspace -v $PWD/data:/workspace/data awml-ptv3
```

### 2. Train
Expand Down
3 changes: 2 additions & 1 deletion projects/TransFusion/Dockerfile
View file
Open in desktop
Original file line number Diff line number Diff line change
@@ -1,9 +1,10 @@
ARG AWML_BASE_IMAGE="ghcr.io/tier4/awml:base-latest"
FROM ${AWML_BASE_IMAGE}
ARG TRT_VERSION=10.8.0.43

# Install pip dependencies (1)
RUN python3 -m pip --no-cache-dir install \
tensorrt
tensorrt-cu12==${TRT_VERSION}

# Install pip dependencies (depend on 1)
RUN python3 -m pip --no-cache-dir install \
Expand Down
4 changes: 2 additions & 2 deletions projects/TransFusion/README.md
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -33,13 +33,13 @@
- Docker build for TransFusion

```sh
DOCKER_BUILDKIT=1 docker build -t autoware-ml-transfusion projects/TransFusion/
DOCKER_BUILDKIT=1 docker build -t awml-transfusion projects/TransFusion/
```

- Run docker

```sh
docker run -it --rm --gpus all --shm-size=64g --name awml -p 6006:6006 -v $PWD/:/workspace -v $PWD/data:/workspace/data autoware-ml-transfusion
docker run -it --rm --gpus all --shm-size=64g --name awml -p 6006:6006 -v $PWD/:/workspace -v $PWD/data:/workspace/data awml-transfusion
```

- Build and install dependency (required only at first run).
Expand Down
4 changes: 2 additions & 2 deletions tools/auto_labeling_3d/Dockerfile
View file
Open in desktop
Original file line number Diff line number Diff line change
@@ -1,5 +1,5 @@
ARG BASE_IMAGE=ghcr.io/tier4/awml:base-latest
FROM ${BASE_IMAGE}
ARG AWML_BASE_IMAGE=ghcr.io/tier4/awml:base-latest
FROM ${AWML_BASE_IMAGE}

# Install pip dependencies
RUN pip install python-json-logger filterpy==1.4.5
Expand Down
2 changes: 1 addition & 1 deletion tools/auto_labeling_3d/README.md
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -64,7 +64,7 @@ graph LR
### Build docker image

- Build docker image.
- If you build `AWML` image locally, please add `--build-arg BASE_IMAGE=autoware-ml` or `--build-arg BASE_IMAGE=autoware-ml-ros2` to build script.
- If you build `AWML` image locally, please add `--build-arg BASE_IMAGE=awml` or `--build-arg BASE_IMAGE=awml-ros2` to build script.

```sh
DOCKER_BUILDKIT=1 docker build -t auto_labeling_3d -f tools/auto_labeling_3d/Dockerfile .
Expand Down
2 changes: 1 addition & 1 deletion tools/scene_selector/README.md
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -18,7 +18,7 @@
### 1. setup environment

```
DOCKER_BUILDKIT=1 docker build -t autoware-ml-ros2 ./tools/setting_environment/ros2/
DOCKER_BUILDKIT=1 docker build -t awml-ros2 ./tools/setting_environment/ros2/
```

### 2. select scene
Expand Down
3 changes: 2 additions & 1 deletion tools/setting_environment/tensorrt/Dockerfile
View file
Open in desktop
Original file line number Diff line number Diff line change
@@ -1,4 +1,5 @@
ARG AWML_BASE_IMAGE="ghcr.io/tier4/awml:base-latest"
FROM ${AWML_BASE_IMAGE}
ARG TRT_VERSION=10.8.0.43

RUN python3 -m pip --no-cache-dir install tensorrt==10.8.0.43 pycuda==2025.1
RUN python3 -m pip --no-cache-dir install tensorrt==${TRT_VERSION} pycuda