chore(docker): add support for multiple Linux distributions and CUDA versions

README.md

@@ -92,6 +92,15 @@ Other projects related to using Rust on the GPU:
 cargo build
 ```
 
+## Use Rust-CUDA in Container Environments
+
+```bash
+# The distribution related Dockerfile are located in `container` folder.
+# Taking ubuntu 24.04 as an example, run the following command in repository root:
+docker build -f ./container/ubuntu24/Dockerfile -t rust-cuda-ubuntu24 .
+docker run --rm --runtime=nvidia --gpus all -it rust-cuda-ubuntu24
+```
+
 ## License
 
 Licensed under either of

container/rockylinux9-cuda12/Dockerfile

@@ -0,0 +1,31 @@
+FROM nvidia/cuda:12.8.1-cudnn-devel-rockylinux9
+
+RUN dnf -y install \
+    openssl-devel \
+    pkgconfig \
+    redhat-rpm-config \
+    which \
+    xz \
+    zlib-devel
+
+# Get LLVM 7 & libffi.so.6
+WORKDIR /data/llvm7
+RUN curl -sSf -L -O https://dl.fedoraproject.org/pub/epel/9/Everything/x86_64/Packages/l/libffi3.1-3.1-36.el9.x86_64.rpm
+RUN curl -sSf -L -O https://dl.fedoraproject.org/pub/epel/8/Everything/x86_64/Packages/l/llvm7.0-7.0.1-7.el8.x86_64.rpm
+RUN curl -sSf -L -O https://dl.fedoraproject.org/pub/epel/8/Everything/x86_64/Packages/l/llvm7.0-devel-7.0.1-7.el8.x86_64.rpm
+RUN curl -sSf -L -O https://dl.fedoraproject.org/pub/epel/8/Everything/x86_64/Packages/l/llvm7.0-libs-7.0.1-7.el8.x86_64.rpm
+RUN curl -sSf -L -O https://dl.fedoraproject.org/pub/epel/8/Everything/x86_64/Packages/l/llvm7.0-static-7.0.1-7.el8.x86_64.rpm
+RUN dnf -y install ./*.rpm
+RUN ln -s /usr/bin/llvm-config-7-64 /usr/bin/llvm-config
+
+# Get Rust
+RUN curl -sSf -L https://sh.rustup.rs | bash -s -- -y
+ENV PATH="/root/.cargo/bin:${PATH}"
+
+# Setup the workspace
+ADD . /data/Rust-CUDA
+WORKDIR /data/Rust-CUDA
+RUN rustup show
+
+ENV LLVM_LINK_STATIC=1
+ENV RUST_LOG=info

container/ubuntu20-cuda11/Dockerfile

@@ -0,0 +1,32 @@
+FROM nvidia/cuda:11.2.2-cudnn8-devel-ubuntu20.04
+
+RUN apt-get update
+RUN DEBIAN_FRONTEND=noninteractive apt-get -qq -y install \
+    build-essential \
+    curl \
+    libssl-dev \
+    libtinfo-dev \
+    pkg-config \
+    xz-utils \
+    zlib1g-dev
+
+# Get LLVM 7
+WORKDIR /data/llvm7
+RUN curl -sSf -L -O http://mirrors.kernel.org/ubuntu/pool/universe/l/llvm-toolchain-7/llvm-7_7.0.1-12_amd64.deb
+RUN curl -sSf -L -O http://mirrors.kernel.org/ubuntu/pool/universe/l/llvm-toolchain-7/llvm-7-dev_7.0.1-12_amd64.deb
+RUN curl -sSf -L -O http://mirrors.kernel.org/ubuntu/pool/universe/l/llvm-toolchain-7/libllvm7_7.0.1-12_amd64.deb
+RUN curl -sSf -L -O http://mirrors.kernel.org/ubuntu/pool/universe/l/llvm-toolchain-7/llvm-7-runtime_7.0.1-12_amd64.deb
+RUN apt-get install -y ./*.deb
+RUN ln -s /usr/bin/llvm-config-7 /usr/bin/llvm-config
+
+# Get Rust
+RUN curl -sSf -L https://sh.rustup.rs | bash -s -- -y
+ENV PATH="/root/.cargo/bin:${PATH}"
+
+# Setup the workspace
+ADD . /data/Rust-CUDA
+WORKDIR /data/Rust-CUDA
+RUN rustup show
+
+ENV LLVM_LINK_STATIC=1
+ENV RUST_LOG=info

container/ubuntu20-cuda12/Dockerfile

@@ -0,0 +1,32 @@
+FROM nvidia/cuda:12.8.1-cudnn-devel-ubuntu20.04
+
+RUN apt-get update
+RUN DEBIAN_FRONTEND=noninteractive apt-get -qq -y install \
+    build-essential \
+    curl \
+    libssl-dev \
+    libtinfo-dev \
+    pkg-config \
+    xz-utils \
+    zlib1g-dev
+
+# Get LLVM 7
+WORKDIR /data/llvm7
+RUN curl -sSf -L -O http://mirrors.kernel.org/ubuntu/pool/universe/l/llvm-toolchain-7/llvm-7_7.0.1-12_amd64.deb
+RUN curl -sSf -L -O http://mirrors.kernel.org/ubuntu/pool/universe/l/llvm-toolchain-7/llvm-7-dev_7.0.1-12_amd64.deb
+RUN curl -sSf -L -O http://mirrors.kernel.org/ubuntu/pool/universe/l/llvm-toolchain-7/libllvm7_7.0.1-12_amd64.deb
+RUN curl -sSf -L -O http://mirrors.kernel.org/ubuntu/pool/universe/l/llvm-toolchain-7/llvm-7-runtime_7.0.1-12_amd64.deb
+RUN apt-get install -y ./*.deb
+RUN ln -s /usr/bin/llvm-config-7 /usr/bin/llvm-config
+
+# Get Rust
+RUN curl -sSf -L https://sh.rustup.rs | bash -s -- -y
+ENV PATH="/root/.cargo/bin:${PATH}"
+
+# Setup the workspace
+ADD . /data/Rust-CUDA
+WORKDIR /data/Rust-CUDA
+RUN rustup show
+
+ENV LLVM_LINK_STATIC=1
+ENV RUST_LOG=info

container/ubuntu24-cuda12/Dockerfile

@@ -0,0 +1,33 @@
+FROM nvidia/cuda:12.8.1-cudnn-devel-ubuntu24.04
+
+RUN apt-get update
+RUN DEBIAN_FRONTEND=noninteractive apt-get -qq -y install \
+    build-essential \
+    curl \
+    libssl-dev \
+    libtinfo-dev \
+    pkg-config \
+    xz-utils \
+    zlib1g-dev
+
+# Get LLVM 7 & libffi7
+WORKDIR /data/llvm7
+RUN curl -sSf -L -O http://security.ubuntu.com/ubuntu/pool/universe/libf/libffi7/libffi7_3.3-5ubuntu1_amd64.deb
+RUN curl -sSf -L -O http://mirrors.kernel.org/ubuntu/pool/universe/l/llvm-toolchain-7/llvm-7_7.0.1-12_amd64.deb
+RUN curl -sSf -L -O http://mirrors.kernel.org/ubuntu/pool/universe/l/llvm-toolchain-7/llvm-7-dev_7.0.1-12_amd64.deb
+RUN curl -sSf -L -O http://mirrors.kernel.org/ubuntu/pool/universe/l/llvm-toolchain-7/libllvm7_7.0.1-12_amd64.deb
+RUN curl -sSf -L -O http://mirrors.kernel.org/ubuntu/pool/universe/l/llvm-toolchain-7/llvm-7-runtime_7.0.1-12_amd64.deb
+RUN apt-get install -y ./*.deb
+RUN ln -s /usr/bin/llvm-config-7 /usr/bin/llvm-config
+
+# Get Rust
+RUN curl -sSf -L https://sh.rustup.rs | bash -s -- -y
+ENV PATH="/root/.cargo/bin:${PATH}"
+
+# Setup the workspace
+ADD . /data/Rust-CUDA
+WORKDIR /data/Rust-CUDA
+RUN rustup show
+
+ENV LLVM_LINK_STATIC=1
+ENV RUST_LOG=info

crates/rustc_codegen_nvvm/build.rs

@@ -350,8 +350,9 @@ fn rustc_llvm_build() {
 
 #[cfg(not(target_os = "windows"))]
 fn link_llvm_system_libs(llvm_config: &Path, components: &[&str]) {
-    let mut cmd = Command::new(llvm_config);
-    cmd.arg("--system-libs");
+    let (_, llvm_link_arg) = detect_llvm_link();
+    let mut cmd: Command = Command::new(llvm_config);
+    cmd.arg(llvm_link_arg).arg("--system-libs");
 
     for comp in components {
         cmd.arg(comp);