missing cuda lib

Signed-off-by: Ryan Cook <[email protected]>
opendatahub-io · Jan 3, 2025 · 64e9247 · 64e9247
1 parent fdcbb9f
commit 64e9247
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/rhoai-ilab-image/Containerfile b/rhoai-ilab-image/Containerfile
@@ -1,6 +1,10 @@
 FROM nvcr.io/nvidia/cuda:12.6.3-cudnn-devel-ubi9
 
-RUN dnf update -y && dnf clean all
+RUN dnf -y install openblas-devel git python3-pip && \
+    dnf install -y libcudnn8 nvidia-driver-NVML nvidia-driver-cuda-libs && \
+    dnf clean all
 
+RUN pip cache remove llama_cpp_python
 RUN pip install kfp==2.9.0
 RUN pip install kubeflow-training
+RUN CMAKE_ARGS="-DLLAMA_CUDA=on -DLLAMA_NATIVE=off" pip install 'instructlab[cuda]'