Experimental: Build default Docker images for both Ubuntu versions 20.04 and 22.04

Author: peterschmidt85

.github/workflows/docker.yml

@@ -78,6 +78,7 @@ jobs:
             --platform linux/amd64 \
             --tag dstackai/${{ env.BUILD_DOCKER_REPO }}:${{ inputs.image_version }}-${{ matrix.flavor }}-ubuntu${{ matrix.ubuntu_version }}.04 \
             --build-arg FLAVOR=${{ matrix.flavor }} \
+            --build-arg UBUNTU_VERSION=${{ matrix.ubuntu_version }} \
             --provenance=false \
             --push \
             -f $FILE .

docker/base/Dockerfile

@@ -53,8 +53,6 @@ RUN cd /opt \
 
 # Final stage
 
-ARG UBUNTU_VERSION
-
 INCLUDE+ base/Dockerfile.common
 
 ENV NCCL_HOME=/opt/nccl

docker/base/efa/Dockerfile

@@ -1,7 +1,5 @@
 # syntax = edrevo/dockerfile-plus
 
-ARG UBUNTU_VERSION
-
 INCLUDE+ base/Dockerfile.common
 
 ENV NCCL_HOME=/usr/local