feat: update cuda to 12.4.1

This commit is contained in:
Manuel Schmid 2024-05-22 23:51:19 +02:00
parent 7f6dfe25a5
commit 260193e36f
No known key found for this signature in database
GPG Key ID: 32C4F7569B40B84B
1 changed files with 1 additions and 4 deletions

View File

@ -1,4 +1,4 @@
FROM nvidia/cuda:12.4.0-base-ubuntu22.04
FROM nvidia/cuda:12.4.1-base-ubuntu22.04
ENV DEBIAN_FRONTEND noninteractive
ENV CMDARGS --listen
@ -14,9 +14,6 @@ RUN pip install --no-cache-dir xformers==0.0.23 --no-dependencies
RUN curl -fsL -o /usr/local/lib/python3.10/dist-packages/gradio/frpc_linux_amd64_v0.2 https://cdn-media.huggingface.co/frpc-gradio-0.2/frpc_linux_amd64 && \
chmod +x /usr/local/lib/python3.10/dist-packages/gradio/frpc_linux_amd64_v0.2
# fix Error: libcuda.so: cannot open shared object file:
RUN ln -s /usr/lib64/libcuda.so.550.67 /usr/lib64/libcuda.so
RUN adduser --disabled-password --gecos '' user && \
mkdir -p /content/app /content/data