dosesnrolls1 commited on
Commit
769e703
·
verified ·
1 Parent(s): ceb6930

Update Dockerfile

Browse files
Files changed (1) hide show
  1. 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, but they land inside
25
- # site-packages rather than a
 
 
 
 
 
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"]