update to ROCm 7.1.1
This commit is contained in:
parent
8939acd913
commit
060cba38c5
|
|
@ -4,7 +4,7 @@ ARG FLAVOR=${TARGETARCH}
|
|||
ARG PARALLEL=8
|
||||
|
||||
ARG ROCM6VERSION=6.4.4
|
||||
ARG ROCM7VERSION=7.1
|
||||
ARG ROCM7VERSION=7.1.1
|
||||
ARG JETPACK5VERSION=r35.4.1
|
||||
ARG JETPACK6VERSION=r36.4.0
|
||||
ARG CMAKEVERSION=3.31.2
|
||||
|
|
@ -206,7 +206,7 @@ RUN apt-get update \
|
|||
COPY --from=archive /bin /usr/bin
|
||||
ENV PATH=/usr/local/sbin:/usr/local/bin:/usr/sbin:/usr/bin:/sbin:/bin
|
||||
COPY --from=archive /lib/ollama /usr/lib/ollama
|
||||
ENV LD_LIBRARY_PATH=/usr/local/nvidia/lib:/usr/local/nvidia/lib64:/usr/lib/ollama:/usr/lib/ollama/rocm_v7
|
||||
ENV LD_LIBRARY_PATH=/usr/local/nvidia/lib:/usr/local/nvidia/lib64:/usr/lib/ollama:/usr/lib/ollama/rocm
|
||||
ENV NVIDIA_DRIVER_CAPABILITIES=compute,utility
|
||||
ENV NVIDIA_VISIBLE_DEVICES=all
|
||||
ENV OLLAMA_HOST=0.0.0.0:11434
|
||||
|
|
|
|||
Loading…
Reference in New Issue