Spaces:
Runtime error
Runtime error
Commit ·
b6d7575
1
Parent(s): 13ed859
Fixes for CUDA
Browse files- Dockerfile +4 -2
Dockerfile
CHANGED
|
@@ -1,9 +1,11 @@
|
|
| 1 |
-
FROM nvidia/cuda:11.8.0-devel-ubuntu22.04
|
| 2 |
|
| 3 |
# Install Python 3.10
|
| 4 |
RUN apt-get update && apt-get install -y --no-install-recommends \
|
| 5 |
python3.10 \
|
| 6 |
python3-pip \
|
|
|
|
|
|
|
| 7 |
&& rm -rf /var/lib/apt/lists/*
|
| 8 |
|
| 9 |
WORKDIR /code
|
|
@@ -29,7 +31,7 @@ ENV INSIGHTFACE_HOME=/tmp/.insightface
|
|
| 29 |
ENV MPLCONFIGDIR=/tmp/matplotlib
|
| 30 |
ENV CUDA_HOME=/usr/local/cuda
|
| 31 |
ENV PATH=${CUDA_HOME}/bin:${PATH}
|
| 32 |
-
ENV LD_LIBRARY_PATH=${CUDA_HOME}/lib64:${LD_LIBRARY_PATH}
|
| 33 |
|
| 34 |
EXPOSE 7860
|
| 35 |
|
|
|
|
| 1 |
+
FROM nvidia/cuda:11.8.0-cudnn8-devel-ubuntu22.04
|
| 2 |
|
| 3 |
# Install Python 3.10
|
| 4 |
RUN apt-get update && apt-get install -y --no-install-recommends \
|
| 5 |
python3.10 \
|
| 6 |
python3-pip \
|
| 7 |
+
libcublas-11-8 \
|
| 8 |
+
libcudnn8 \
|
| 9 |
&& rm -rf /var/lib/apt/lists/*
|
| 10 |
|
| 11 |
WORKDIR /code
|
|
|
|
| 31 |
ENV MPLCONFIGDIR=/tmp/matplotlib
|
| 32 |
ENV CUDA_HOME=/usr/local/cuda
|
| 33 |
ENV PATH=${CUDA_HOME}/bin:${PATH}
|
| 34 |
+
ENV LD_LIBRARY_PATH=${CUDA_HOME}/lib64:${CUDA_HOME}/extras/CUPTI/lib64:${LD_LIBRARY_PATH}
|
| 35 |
|
| 36 |
EXPOSE 7860
|
| 37 |
|