From ee8ab99f4878668de90305a9e6b4eef44994111c Mon Sep 17 00:00:00 2001
From: tcsenpai <dev@tcsenpai.com>
Date: Fri, 23 May 2025 13:03:28 +0200
Subject: [PATCH] fixed ctranslate2 related errors

---
 Dockerfile | 8 ++++++++
 1 file changed, 8 insertions(+)

diff --git a/Dockerfile b/Dockerfile
index b8e0617..18bca33 100644
--- a/Dockerfile
+++ b/Dockerfile
@@ -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 \