@@ -42,7 +42,8 @@ check_requirements() {
4242 exit 1
4343 fi
4444
45- if ! command -v docker-compose & > /dev/null; then
45+ # Check for Docker Compose (v2 with 'docker compose' or legacy v1 with 'docker-compose')
46+ if ! docker compose version & > /dev/null && ! command -v docker-compose & > /dev/null; then
4647 error " Docker Compose is not installed. Please install Docker Compose first."
4748 exit 1
4849 fi
@@ -77,18 +78,19 @@ build_platform() {
7778 fi
7879}
7980
80- # Build using docker-compose
81+ # Build using docker compose (v2) or docker -compose (v1)
8182build_compose () {
8283 local service=$1
83- log " Building $service using docker- compose..."
84+ log " Building $service using docker compose..."
8485
8586 cd " $DOCKER_DIR "
8687
87- if docker-compose build " $service " ; then
88- success " $service built successfully using docker-compose"
88+ # Try docker compose (v2) first, then fall back to docker-compose (v1)
89+ if docker compose build " $service " 2> /dev/null || docker-compose build " $service " 2> /dev/null; then
90+ success " $service built successfully using docker compose"
8991 return 0
9092 else
91- error " Failed to build $service using docker- compose"
93+ error " Failed to build $service using docker compose"
9294 return 1
9395 fi
9496}
@@ -103,7 +105,7 @@ Usage: $0 [OPTIONS] [PLATFORM]
103105Options:
104106 -h, --help Show this help message
105107 -a, --all Build all platforms (CUDA + ROCm)
106- -c, --compose Use docker- compose for building
108+ -c, --compose Use docker compose for building
107109 --clean Clean existing images before building
108110 --no-cache Build without using Docker cache
109111 --pull Pull base images before building
@@ -116,7 +118,7 @@ Examples:
116118 $0 cuda Build CUDA container only
117119 $0 rocm Build ROCm container only
118120 $0 --all Build both CUDA and ROCm containers
119- $0 --compose cuda Build using docker- compose
121+ $0 --compose cuda Build using docker compose
120122 $0 --clean --all Clean and rebuild all containers
121123
122124EOF
@@ -246,7 +248,7 @@ main() {
246248 log " Next steps:"
247249 log " - Run: ./docker/scripts/run.sh cuda (for NVIDIA GPUs)"
248250 log " - Run: ./docker/scripts/run.sh rocm (for AMD GPUs)"
249- log " - Or use: docker- compose -f docker/docker-compose.yml up cuda-dev"
251+ log " - Or use: docker compose -f docker/docker-compose.yml up cuda-dev"
250252 else
251253 warning " Some builds failed. Check the logs above for details."
252254 exit 1
0 commit comments