diff --git a/docs/tutorial/tutorial_detection_3d.md b/docs/tutorial/tutorial_detection_3d.md index 4f2728e06..747df942a 100644 --- a/docs/tutorial/tutorial_detection_3d.md +++ b/docs/tutorial/tutorial_detection_3d.md @@ -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 . ``` diff --git a/projects/CalibrationStatusClassification/Dockerfile b/projects/CalibrationStatusClassification/Dockerfile index 77153382f..53a6856a5 100644 --- a/projects/CalibrationStatusClassification/Dockerfile +++ b/projects/CalibrationStatusClassification/Dockerfile @@ -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 diff --git a/projects/CalibrationStatusClassification/README.md b/projects/CalibrationStatusClassification/README.md index 6d8a8694c..ef96ee0eb 100644 --- a/projects/CalibrationStatusClassification/README.md +++ b/projects/CalibrationStatusClassification/README.md @@ -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 diff --git a/projects/FRNet/Dockerfile b/projects/FRNet/Dockerfile index 691db3190..5e8f94a2f 100644 --- a/projects/FRNet/Dockerfile +++ b/projects/FRNet/Dockerfile @@ -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 \ @@ -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 . diff --git a/projects/FRNet/README.md b/projects/FRNet/README.md index f47a08d9c..f1b3cd1af 100644 --- a/projects/FRNet/README.md +++ b/projects/FRNet/README.md @@ -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 diff --git a/projects/PTv3/Dockerfile b/projects/PTv3/Dockerfile index 9a7e0e3ff..178390e4c 100644 --- a/projects/PTv3/Dockerfile +++ b/projects/PTv3/Dockerfile @@ -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" diff --git a/projects/PTv3/README.md b/projects/PTv3/README.md index 3d1e65cb1..06ed1c204 100644 --- a/projects/PTv3/README.md +++ b/projects/PTv3/README.md @@ -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 diff --git a/projects/TransFusion/Dockerfile b/projects/TransFusion/Dockerfile index 5b505e2a8..8d4f88709 100644 --- a/projects/TransFusion/Dockerfile +++ b/projects/TransFusion/Dockerfile @@ -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 \ diff --git a/projects/TransFusion/README.md b/projects/TransFusion/README.md index 16eade0c2..755b3338c 100644 --- a/projects/TransFusion/README.md +++ b/projects/TransFusion/README.md @@ -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). diff --git a/tools/auto_labeling_3d/Dockerfile b/tools/auto_labeling_3d/Dockerfile index 2b291ec58..288a5f8f8 100644 --- a/tools/auto_labeling_3d/Dockerfile +++ b/tools/auto_labeling_3d/Dockerfile @@ -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 diff --git a/tools/auto_labeling_3d/README.md b/tools/auto_labeling_3d/README.md index 3aad534fd..6be6b94f0 100644 --- a/tools/auto_labeling_3d/README.md +++ b/tools/auto_labeling_3d/README.md @@ -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 . diff --git a/tools/scene_selector/README.md b/tools/scene_selector/README.md index cea6a995d..b41ea3b2f 100644 --- a/tools/scene_selector/README.md +++ b/tools/scene_selector/README.md @@ -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 diff --git a/tools/setting_environment/tensorrt/Dockerfile b/tools/setting_environment/tensorrt/Dockerfile index b4b424caa..b2118f116 100644 --- a/tools/setting_environment/tensorrt/Dockerfile +++ b/tools/setting_environment/tensorrt/Dockerfile @@ -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