[UX]: Replace conda with uv in dstack's default Docker image #2625

* Updated CUDA to 12.1.1
* Use NVIDIA's devel image

Author: peterschmidt85

.github/workflows/docker.yml

@@ -60,8 +60,8 @@ jobs:
         uses: docker/setup-qemu-action@v3
       - name: Build and upload to DockerHub
         run: |
-          docker buildx build --platform linux/amd64 --build-arg PYTHON=${{ matrix.python }} --push --provenance=false --tag dstackai/base:py${{ matrix.python }}-${{ inputs.image_version }}-cuda-12.1 -f base/Dockerfile .
-          docker buildx build --platform linux/amd64 --build-arg PYTHON=${{ matrix.python }} --build-arg VERSION=${{ inputs.image_version }} --push --provenance=false --tag dstackai/base:py${{ matrix.python }}-${{ inputs.image_version }}-cuda-12.1-devel -f base/devel.Dockerfile .
+          docker buildx build --platform linux/amd64 --build-arg FLAVOR=base --build-arg PYTHON=${{ matrix.python }} --push --provenance=false --tag dstackai/base:py${{ matrix.python }}-${{ inputs.image_version }}-cuda-12.1 -f base/Dockerfile .
+          docker buildx build --platform linux/amd64 --build-arg FLAVOR=devel --build-arg PYTHON=${{ matrix.python }} --build-arg VERSION=${{ inputs.image_version }} --push --provenance=false --tag dstackai/base:py${{ matrix.python }}-${{ inputs.image_version }}-cuda-12.1-devel -f base/devel.Dockerfile .
 
   build-aws-images:
     needs: build-docker

docker/base/Dockerfile

@@ -1,4 +1,5 @@
-FROM nvidia/cuda:12.1.0-base-ubuntu20.04
+ARG FLAVOR
+FROM nvidia/cuda:12.1.1-${FLAVOR}-ubuntu20.04
 
 ARG PYTHON
 ARG _UV_HOME="/opt/uv"