Install rust in docker

mlc-ai · Jul 11, 2024 · fd5cb1b · fd5cb1b
1 parent 87ff5d0
commit fd5cb1b
Show file tree

Hide file tree

Showing 8 changed files with 50 additions and 0 deletions.
diff --git a/docker/Dockerfile.package-cpu b/docker/Dockerfile.package-cpu
@@ -42,3 +42,7 @@ RUN bash /install/almalinux_install_python_package.sh
 COPY install/almalinux_install_auditwheel.sh /install/almalinux_install_auditwheel.sh
 RUN bash /install/almalinux_install_auditwheel.sh
 ENV AUDITWHEEL_PLAT=manylinux_2_28_x86_64
+
+# Install Rust
+COPY install/almalinux_install_rust.sh /install/almalinux_install_rust.sh
+RUN bash /install/almalinux_install_rust.sh
diff --git a/docker/Dockerfile.package-cu117 b/docker/Dockerfile.package-cu117
@@ -49,3 +49,7 @@ RUN bash /install/almalinux_install_python_package.sh
 COPY install/almalinux_install_auditwheel.sh /install/almalinux_install_auditwheel.sh
 RUN bash /install/almalinux_install_auditwheel.sh
 ENV AUDITWHEEL_PLAT=manylinux_2_28_x86_64
+
+# Install Rust
+COPY install/almalinux_install_rust.sh /install/almalinux_install_rust.sh
+RUN bash /install/almalinux_install_rust.sh
diff --git a/docker/Dockerfile.package-cu118 b/docker/Dockerfile.package-cu118
@@ -49,3 +49,7 @@ RUN bash /install/almalinux_install_python_package.sh
 COPY install/almalinux_install_auditwheel.sh /install/almalinux_install_auditwheel.sh
 RUN bash /install/almalinux_install_auditwheel.sh
 ENV AUDITWHEEL_PLAT=manylinux_2_28_x86_64
+
+# Install Rust
+COPY install/almalinux_install_rust.sh /install/almalinux_install_rust.sh
+RUN bash /install/almalinux_install_rust.sh
diff --git a/docker/Dockerfile.package-cu121 b/docker/Dockerfile.package-cu121
@@ -49,3 +49,7 @@ RUN bash /install/almalinux_install_python_package.sh
 COPY install/almalinux_install_auditwheel.sh /install/almalinux_install_auditwheel.sh
 RUN bash /install/almalinux_install_auditwheel.sh
 ENV AUDITWHEEL_PLAT=manylinux_2_28_x86_64
+
+# Install Rust
+COPY install/almalinux_install_rust.sh /install/almalinux_install_rust.sh
+RUN bash /install/almalinux_install_rust.sh
diff --git a/docker/Dockerfile.package-cu122 b/docker/Dockerfile.package-cu122
@@ -49,3 +49,7 @@ RUN bash /install/almalinux_install_python_package.sh
 COPY install/almalinux_install_auditwheel.sh /install/almalinux_install_auditwheel.sh
 RUN bash /install/almalinux_install_auditwheel.sh
 ENV AUDITWHEEL_PLAT=manylinux_2_28_x86_64
+
+# Install Rust
+COPY install/almalinux_install_rust.sh /install/almalinux_install_rust.sh
+RUN bash /install/almalinux_install_rust.sh
diff --git a/docker/Dockerfile.package-rocm56 b/docker/Dockerfile.package-rocm56
@@ -46,3 +46,7 @@ RUN bash /install/almalinux_install_python_package.sh
 COPY install/almalinux_install_auditwheel.sh /install/almalinux_install_auditwheel.sh
 RUN bash /install/almalinux_install_auditwheel.sh
 ENV AUDITWHEEL_PLAT=manylinux_2_28_x86_64
+
+# Install Rust
+COPY install/almalinux_install_rust.sh /install/almalinux_install_rust.sh
+RUN bash /install/almalinux_install_rust.sh
diff --git a/docker/Dockerfile.package-rocm57 b/docker/Dockerfile.package-rocm57
@@ -46,3 +46,7 @@ RUN bash /install/almalinux_install_python_package.sh
 COPY install/almalinux_install_auditwheel.sh /install/almalinux_install_auditwheel.sh
 RUN bash /install/almalinux_install_auditwheel.sh
 ENV AUDITWHEEL_PLAT=manylinux_2_28_x86_64
+
+# Install Rust
+COPY install/almalinux_install_rust.sh /install/almalinux_install_rust.sh
+RUN bash /install/almalinux_install_rust.sh
diff --git a/docker/install/almalinux_install_rust.sh b/docker/install/almalinux_install_rust.sh
@@ -0,0 +1,22 @@
+#!/bin/bash
+set -euxo pipefail
+
+export RUSTUP_HOME=/opt/rust
+export CARGO_HOME=/opt/rust
+
+# this rustc is one supported by the installed version of rust-sgx-sdk
+HOST_ARG=
+if [ "$(getconf LONG_BIT)" == "32" ]; then
+    # When building in the i386 docker image on a 64-bit host, rustup doesn't
+    # correctly detect the arch to install for so set it manually
+    HOST_ARG="--default-host i686-unknown-linux-gnu"
+fi
+
+# shellcheck disable=SC2086 # word splitting is intentional here
+curl -s -S -L https://sh.rustup.rs -sSf | sh -s -- -y --no-modify-path --profile minimal --default-toolchain stable $HOST_ARG
+export PATH=$CARGO_HOME/bin:$PATH
+rustup component add rustfmt
+rustup component add clippy
+
+# make rust usable by all users after install during container build
+chmod -R a+rw /opt/rust