mirror of
https://github.com/nod-ai/SHARK-Studio.git
synced 2026-01-09 13:57:54 -05:00
89 lines
3.2 KiB
Plaintext
89 lines
3.2 KiB
Plaintext
ARG IMAGE_NAME
|
|
FROM ${IMAGE_NAME}:12.2.0-runtime-ubuntu22.04 as base
|
|
|
|
ENV NV_CUDA_LIB_VERSION "12.2.0-1"
|
|
|
|
FROM base as base-amd64
|
|
|
|
ENV NV_CUDA_CUDART_DEV_VERSION 12.2.53-1
|
|
ENV NV_NVML_DEV_VERSION 12.2.81-1
|
|
ENV NV_LIBCUSPARSE_DEV_VERSION 12.1.1.53-1
|
|
ENV NV_LIBNPP_DEV_VERSION 12.1.1.14-1
|
|
ENV NV_LIBNPP_DEV_PACKAGE libnpp-dev-12-2=${NV_LIBNPP_DEV_VERSION}
|
|
|
|
ENV NV_LIBCUBLAS_DEV_VERSION 12.2.1.16-1
|
|
ENV NV_LIBCUBLAS_DEV_PACKAGE_NAME libcublas-dev-12-2
|
|
ENV NV_LIBCUBLAS_DEV_PACKAGE ${NV_LIBCUBLAS_DEV_PACKAGE_NAME}=${NV_LIBCUBLAS_DEV_VERSION}
|
|
|
|
ENV NV_CUDA_NSIGHT_COMPUTE_VERSION 12.2.0-1
|
|
ENV NV_CUDA_NSIGHT_COMPUTE_DEV_PACKAGE cuda-nsight-compute-12-2=${NV_CUDA_NSIGHT_COMPUTE_VERSION}
|
|
|
|
ENV NV_NVPROF_VERSION 12.2.60-1
|
|
ENV NV_NVPROF_DEV_PACKAGE cuda-nvprof-12-2=${NV_NVPROF_VERSION}
|
|
FROM base as base-arm64
|
|
|
|
ENV NV_CUDA_CUDART_DEV_VERSION 12.2.53-1
|
|
ENV NV_NVML_DEV_VERSION 12.2.81-1
|
|
ENV NV_LIBCUSPARSE_DEV_VERSION 12.1.1.53-1
|
|
ENV NV_LIBNPP_DEV_VERSION 12.1.1.14-1
|
|
ENV NV_LIBNPP_DEV_PACKAGE libnpp-dev-12-2=${NV_LIBNPP_DEV_VERSION}
|
|
|
|
ENV NV_LIBCUBLAS_DEV_PACKAGE_NAME libcublas-dev-12-2
|
|
ENV NV_LIBCUBLAS_DEV_VERSION 12.2.1.16-1
|
|
ENV NV_LIBCUBLAS_DEV_PACKAGE ${NV_LIBCUBLAS_DEV_PACKAGE_NAME}=${NV_LIBCUBLAS_DEV_VERSION}
|
|
|
|
ENV NV_CUDA_NSIGHT_COMPUTE_VERSION 12.2.0-1
|
|
ENV NV_CUDA_NSIGHT_COMPUTE_DEV_PACKAGE cuda-nsight-compute-12-2=${NV_CUDA_NSIGHT_COMPUTE_VERSION}
|
|
|
|
FROM base-${TARGETARCH}
|
|
|
|
ARG TARGETARCH
|
|
|
|
LABEL maintainer "AMDSHARK<stdin@nod.com>"
|
|
|
|
# Register the ROCM package repository, and install rocm-dev package
|
|
ARG ROCM_VERSION=5.6
|
|
ARG AMDGPU_VERSION=5.6
|
|
|
|
ARG APT_PREF
|
|
RUN echo "$APT_PREF" > /etc/apt/preferences.d/rocm-pin-600
|
|
RUN apt-get update && DEBIAN_FRONTEND=noninteractive apt-get install -y --no-install-recommends ca-certificates curl libnuma-dev gnupg \
|
|
&& curl -sL https://repo.radeon.com/rocm/rocm.gpg.key | apt-key add - \
|
|
&& printf "deb [arch=amd64] https://repo.radeon.com/rocm/apt/$ROCM_VERSION/ jammy main" | tee /etc/apt/sources.list.d/rocm.list \
|
|
&& printf "deb [arch=amd64] https://repo.radeon.com/amdgpu/$AMDGPU_VERSION/ubuntu jammy main" | tee /etc/apt/sources.list.d/amdgpu.list \
|
|
&& apt-get update && DEBIAN_FRONTEND=noninteractive apt-get install -y --no-install-recommends \
|
|
sudo \
|
|
libelf1 \
|
|
kmod \
|
|
file \
|
|
python3 \
|
|
python3-pip \
|
|
rocm-dev \
|
|
rocm-libs \
|
|
rocm-hip-libraries \
|
|
build-essential && \
|
|
apt-get clean && \
|
|
rm -rf /var/lib/apt/lists/*
|
|
|
|
RUN groupadd -g 109 render
|
|
|
|
RUN apt-get update && apt-get install -y --no-install-recommends \
|
|
cuda-cudart-dev-12-2=${NV_CUDA_CUDART_DEV_VERSION} \
|
|
cuda-command-line-tools-12-2=${NV_CUDA_LIB_VERSION} \
|
|
cuda-minimal-build-12-2=${NV_CUDA_LIB_VERSION} \
|
|
cuda-libraries-dev-12-2=${NV_CUDA_LIB_VERSION} \
|
|
cuda-nvml-dev-12-2=${NV_NVML_DEV_VERSION} \
|
|
${NV_NVPROF_DEV_PACKAGE} \
|
|
${NV_LIBNPP_DEV_PACKAGE} \
|
|
libcusparse-dev-12-2=${NV_LIBCUSPARSE_DEV_VERSION} \
|
|
${NV_LIBCUBLAS_DEV_PACKAGE} \
|
|
${NV_CUDA_NSIGHT_COMPUTE_DEV_PACKAGE} \
|
|
&& rm -rf /var/lib/apt/lists/*
|
|
|
|
RUN apt install rocm-hip-libraries
|
|
|
|
# Keep apt from auto upgrading the cublas and nccl packages. See https://gitlab.com/nvidia/container-images/cuda/-/issues/88
|
|
RUN apt-mark hold ${NV_LIBCUBLAS_DEV_PACKAGE_NAME}
|
|
ENV LIBRARY_PATH /usr/local/cuda/lib64/stubs
|
|
|