[Feature]: Support for Tenstorrent Galaxy #2898 (WIP)

Added dstackai/tt-smi Dockerfile + CI/DI workflow

Author: peterschmidt85

.github/workflows/docker-tt-smi.yml

@@ -0,0 +1,45 @@
+name: Build TT SMI Docker image
+
+on:
+  workflow_dispatch:
+    inputs:
+      image_name:
+        description: "Docker image name"
+        required: true
+        default: "dstackai/tt-smi"
+      tt_smi_version:
+        description: "TT SMI version"
+        required: true
+        default: "3.0.25"
+
+jobs:
+  build-tt-smi:
+    defaults:
+      run:
+        working-directory: docker/tt-smi
+    runs-on: ubuntu-latest
+    steps:
+      - name: Checkout repository
+        uses: actions/checkout@v4
+      - name: Set up Docker Buildx
+        uses: docker/setup-buildx-action@v3
+      - name: Login to DockerHub
+        uses: docker/login-action@v3
+        with:
+          username: ${{ secrets.DOCKERHUB_USERNAME }}
+          password: ${{ secrets.DOCKERHUB_TOKEN }}
+      - name: Build and upload to DockerHub
+        run: |
+          IMAGE_NAME=${{ inputs.image_name }}
+          docker buildx build . \
+            --load \
+            --provenance=false \
+            --platform linux/amd64 \
+            --build-arg IMAGE_NAME=${IMAGE_NAME} \
+            --build-arg TT_SMI_VERSION=${{ inputs.tt_smi_version }} \
+            --build-arg BUILD_DATE=$(date --utc --iso-8601=seconds)Z \
+            --tag ${IMAGE_NAME}:latest
+          VERSION=$(docker inspect --format '{{ index .Config.Labels "org.opencontainers.image.version" }}' ${IMAGE_NAME})
+          docker tag ${IMAGE_NAME}:latest ${IMAGE_NAME}:${VERSION}
+          docker push ${IMAGE_NAME}:${VERSION}
+          docker push ${IMAGE_NAME}:latest

docker/tt-smi/Dockerfile

@@ -0,0 +1,22 @@
+FROM ubuntu:20.04
+
+ARG IMAGE_NAME
+ARG TT_SMI_VERSION
+ARG BUILD_DATE
+
+ENV PATH="/root/.cargo/bin:${PATH}"
+
+RUN \
+    apt-get update && \
+    apt-get install -y curl git python3 python3-pip && \
+    curl --proto '=https' --tlsv1.2 -sSf https://sh.rustup.rs | sh -s -- -y && \
+    pip3 install --no-cache-dir git+https://github.com/tenstorrent/tt-smi@v${TT_SMI_VERSION} && \
+    apt-get clean && \
+    rm -rf /var/lib/apt/lists/*
+
+ENTRYPOINT ["/usr/local/bin/tt-smi"]
+CMD ["--help"]
+
+LABEL org.opencontainers.image.title="${IMAGE_NAME}"
+LABEL org.opencontainers.image.version="${TT_SMI_VERSION}"
+LABEL org.opencontainers.image.created="${BUILD_DATE}"

docker/tt-smi/README.md

@@ -0,0 +1,9 @@
+# dstack TT SMI
+
+An Ubuntu-based image with [TT SMI](https://github.com/tenstorrent/tt-smi/) preinstalled. Suitable for Tenstorrent GPU detection.
+
+## Usage
+
+```shell
+docker run --device /dev/tenstorrent/ dstackai/tt-smi -s
+```