Spaces:
Paused
Paused
Update Dockerfile
Browse files- Dockerfile +6 -2
Dockerfile
CHANGED
|
@@ -21,5 +21,9 @@ ENV GRADIO_SERVER_PORT=7860
|
|
| 21 |
EXPOSE 7860
|
| 22 |
|
| 23 |
# The pip-installed nvidia-*-cu12 packages (see requirements.txt) ship the
|
| 24 |
-
# actual CUDA/cuDNN .so files onnxruntime-gpu needs
|
| 25 |
-
#
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 21 |
EXPOSE 7860
|
| 22 |
|
| 23 |
# The pip-installed nvidia-*-cu12 packages (see requirements.txt) ship the
|
| 24 |
+
# actual CUDA/cuDNN .so files onnxruntime-gpu needs. python:3.10-slim always
|
| 25 |
+
# installs them to this exact path, so we hardcode it rather than computing
|
| 26 |
+
# it dynamically at container start.
|
| 27 |
+
ENV LD_LIBRARY_PATH="/usr/local/lib/python3.10/site-packages/nvidia/cublas/lib:/usr/local/lib/python3.10/site-packages/nvidia/cudnn/lib:/usr/local/lib/python3.10/site-packages/nvidia/cuda_runtime/lib:/usr/local/lib/python3.10/site-packages/nvidia/cuda_nvrtc/lib:/usr/local/lib/python3.10/site-packages/nvidia/cufft/lib:/usr/local/lib/python3.10/site-packages/nvidia/curand/lib:/usr/local/lib/python3.10/site-packages/nvidia/cusolver/lib:/usr/local/lib/python3.10/site-packages/nvidia/cusparse/lib:${LD_LIBRARY_PATH}"
|
| 28 |
+
|
| 29 |
+
CMD ["python", "app.py"]
|