f844b33b2d
- All files under *core/iwasm/libraries/wasi-nn* are compiled as shared libraries - *wasi-nn.c* is shared between backends - Every backend has a separated shared library - If wasi-nn feature is enabled, iwasm will depend on shared library libiwasm.so instead of linking static library libvmlib.a
48 lines
1.6 KiB
Docker
48 lines
1.6 KiB
Docker
# Copyright (C) 2019 Intel Corporation. All rights reserved.
|
|
# SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
|
|
|
|
FROM ubuntu:20.04 AS base
|
|
|
|
ENV DEBIAN_FRONTEND=noninteractive
|
|
|
|
# hadolint ignore=DL3008,DL3009
|
|
RUN apt-get update \
|
|
&& apt-get install -y --no-install-recommends\
|
|
ca-certificates cmake build-essential git wget
|
|
|
|
WORKDIR /usr/local/share/ca-certificates/cacert.org
|
|
RUN wget -qP /usr/local/share/ca-certificates/cacert.org http://www.cacert.org/certs/root.crt http://www.cacert.org/certs/class3.crt \
|
|
&& update-ca-certificates
|
|
|
|
WORKDIR /home/wamr
|
|
COPY . .
|
|
RUN git config --global http.sslCAinfo /etc/ssl/certs/ca-certificates.crt
|
|
|
|
WORKDIR /home/wamr/product-mini/platforms/linux/build
|
|
RUN rm -rf build \
|
|
&& cmake -S . -B build \
|
|
-DWAMR_BUILD_WASI_NN=1 \
|
|
-DWAMR_BUILD_WASI_NN_ENABLE_GPU=1 \
|
|
&& cmake --build build -j "$(grep -c ^processor /proc/cpuinfo)"
|
|
|
|
FROM nvidia/cuda:11.3.0-runtime-ubuntu20.04
|
|
|
|
# hadolint ignore=DL3008
|
|
RUN apt-get update && apt-get install -y --no-install-recommends \
|
|
ocl-icd-libopencl1 \
|
|
ocl-icd-opencl-dev \
|
|
clinfo && \
|
|
rm -rf /var/lib/apt/lists/*
|
|
|
|
RUN mkdir -p /etc/OpenCL/vendors && \
|
|
echo "libnvidia-opencl.so.1" > /etc/OpenCL/vendors/nvidia.icd
|
|
|
|
ENV NVIDIA_VISIBLE_DEVICES=all
|
|
ENV NVIDIA_DRIVER_CAPABILITIES=compute,utility
|
|
|
|
COPY --from=base /home/wamr/product-mini/platforms/linux/build/iwasm /usr/bin/iwasm
|
|
COPY --from=base /home/wamr/product-mini/platforms/linux/build/libiwasm.so /lib/libiwasm.so
|
|
COPY --from=base /home/wamr/product-mini/platforms/linux/build/libwasi-nn-*.so /lib/
|
|
|
|
ENTRYPOINT [ "iwasm" ]
|