mirror of
https://github.com/vosen/ZLUDA.git
synced 2025-04-20 08:24:44 +00:00
52 lines
2 KiB
Docker
52 lines
2 KiB
Docker
FROM nvidia/cuda:12.4.1-base-ubuntu22.04
|
|
|
|
RUN DEBIAN_FRONTEND=noninteractive apt-get update -y && DEBIAN_FRONTEND=noninteractive apt-get install -y --no-install-recommends \
|
|
wget \
|
|
build-essential \
|
|
cmake \
|
|
ninja-build \
|
|
python3 \
|
|
ripgrep \
|
|
git \
|
|
ltrace \
|
|
# required by llvm 17
|
|
lsb-release software-properties-common gnupg
|
|
|
|
ARG LLVM_VERSION=17
|
|
RUN wget https://apt.llvm.org/llvm.sh && \
|
|
chmod +x llvm.sh && \
|
|
./llvm.sh ${LLVM_VERSION}
|
|
|
|
# Feel free to change to a newer version if you have a newer verison on your host
|
|
ARG CUDA_PKG_VERSION=12-4
|
|
# Docker <-> host driver version compatiblity is newer host <-> older docker
|
|
# We don't care about a specific driver version, so pick oldest 5XX
|
|
ARG CUDA_DRIVER=515
|
|
RUN DEBIAN_FRONTEND=noninteractive apt-get update -y && DEBIAN_FRONTEND=noninteractive apt-get install -y --no-install-recommends \
|
|
# CUDA headers need it for interop
|
|
libgl-dev libegl-dev libvdpau-dev \
|
|
nvidia-utils-${CUDA_DRIVER} \
|
|
cuda-cudart-dev-${CUDA_PKG_VERSION} \
|
|
cuda-nvml-dev-${CUDA_PKG_VERSION} \
|
|
cuda-cudart-${CUDA_PKG_VERSION} \
|
|
cuda-profiler-api-${CUDA_PKG_VERSION} \
|
|
cuda-nvcc-${CUDA_PKG_VERSION}
|
|
|
|
ARG ROCM_VERSION=6.3.1
|
|
RUN mkdir --parents --mode=0755 /etc/apt/keyrings && \
|
|
wget https://repo.radeon.com/rocm/rocm.gpg.key -O - | \
|
|
gpg --dearmor | tee /etc/apt/keyrings/rocm.gpg > /dev/null && \
|
|
echo "deb [arch=amd64 signed-by=/etc/apt/keyrings/rocm.gpg] https://repo.radeon.com/rocm/apt/${ROCM_VERSION} jammy main" > /etc/apt/sources.list.d/rocm.list && \
|
|
echo 'Package: *\nPin: release o=repo.radeon.com\nPin-Priority: 600' > /etc/apt/preferences.d/rocm-pin-600 && \
|
|
DEBIAN_FRONTEND=noninteractive apt update -y && DEBIAN_FRONTEND=noninteractive apt-get install -y --no-install-recommends \
|
|
rocminfo \
|
|
rocm-gdb \
|
|
rocm-smi-lib \
|
|
rocm-llvm-dev \
|
|
hip-runtime-amd \
|
|
hip-dev && \
|
|
echo '/opt/rocm/lib' > /etc/ld.so.conf.d/rocm.conf && \
|
|
ldconfig
|
|
|
|
ENV PATH=$PATH:/opt/rocm-${ROCM_VERSION}/bin
|
|
|