mirror of
https://github.com/tcsenpai/whisperapp.git
synced 2025-06-03 22:00:03 +00:00
fixed ctranslate2 related errors
This commit is contained in:
parent
053e0a6fe4
commit
ee8ab99f48
@ -4,6 +4,14 @@ FROM nvidia/cuda:12.3.2-cudnn9-devel-ubuntu22.04
|
||||
ENV DEBIAN_FRONTEND=noninteractive
|
||||
ENV PYTHONUNBUFFERED=1
|
||||
|
||||
# Define the path to PyTorch's bundled NVIDIA libraries (adjust if necessary for your specific Python version/setup)
|
||||
# This path assumes nvidia-cudnn-cuXX or similar packages install here.
|
||||
ENV PYTORCH_NVIDIA_LIBS_DIR /usr/local/lib/python3.10/dist-packages/nvidia/cudnn/lib
|
||||
# Prepend PyTorch's NVIDIA library directory to LD_LIBRARY_PATH
|
||||
# Also include the standard NVIDIA paths that the base image might set for other CUDA components.
|
||||
ENV LD_LIBRARY_PATH=${PYTORCH_NVIDIA_LIBS_DIR}:/usr/local/nvidia/lib:/usr/local/nvidia/lib64
|
||||
|
||||
|
||||
# Install system dependencies
|
||||
RUN apt-get update && apt-get install -y \
|
||||
python3.10 \
|
||||
|
Loading…
x
Reference in New Issue
Block a user