diff --git a/Dockerfile b/Dockerfile index 6f4e0e9..843ec6e 100644 --- a/Dockerfile +++ b/Dockerfile @@ -7,7 +7,7 @@ RUN dnf -y install --setopt=install_weak_deps=False --nodocs \ gcc gcc-c++ binutils make ffmpeg-free \ cmake ninja-build aria2c tar xz vim nano dialog \ libdrm-devel zlib-devel openssl-devel pgrep \ - numactl-devel gperftools-libs \ + numactl-devel gperftools-libs iproute libibverbs-utils \ && dnf clean all && rm -rf /var/cache/dnf/* # 2. Install "TheRock" ROCm SDK (Tarball Method) @@ -123,6 +123,8 @@ RUN export HIP_DEVICE_LIB_PATH=$(find /opt/rocm -type d -name bitcode -print -qu python -m pip wheel --no-build-isolation --no-deps -w /tmp/dist -v . && \ python -m pip install /tmp/dist/*.whl +RUN python -m pip install ray + # --- bitsandbytes (ROCm) --- WORKDIR /opt RUN git clone -b rocm_enabled_multi_backend https://github.com/ROCm/bitsandbytes.git