Skip to content

Commit b7b7a0a

Browse files
committed
Update docker compose
1 parent 6e97628 commit b7b7a0a

File tree

1 file changed

+11
-9
lines changed

1 file changed

+11
-9
lines changed

docker/scripts/build.sh

Lines changed: 11 additions & 9 deletions
Original file line numberDiff line numberDiff line change
@@ -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)
8182
build_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]
103105
Options:
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
122124
EOF
@@ -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

Comments
 (0)