fix(agent): cuDNN base image so onnxruntime-gpu loads #146
+4
-2
@@ -1,6 +1,8 @@
|
|||||||
# FabledCurator GPU agent — runs on the desktop with the GPU.
|
# FabledCurator GPU agent — runs on the desktop with the GPU.
|
||||||
# CUDA runtime so onnxruntime-gpu can use the card; ffmpeg for video frames.
|
# CUDA + cuDNN runtime so onnxruntime-gpu can use the card (it needs cuDNN 9 —
|
||||||
FROM nvidia/cuda:12.4.1-runtime-ubuntu22.04
|
# the plain -runtime image lacks it: "libcudnn.so.9: cannot open shared object
|
||||||
|
# file"); ffmpeg for video frames.
|
||||||
|
FROM nvidia/cuda:12.4.1-cudnn-runtime-ubuntu22.04
|
||||||
|
|
||||||
ENV DEBIAN_FRONTEND=noninteractive PYTHONUNBUFFERED=1
|
ENV DEBIAN_FRONTEND=noninteractive PYTHONUNBUFFERED=1
|
||||||
RUN apt-get update \
|
RUN apt-get update \
|
||||||
|
|||||||
Reference in New Issue
Block a user