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.
|
||||
# CUDA runtime so onnxruntime-gpu can use the card; ffmpeg for video frames.
|
||||
FROM nvidia/cuda:12.4.1-runtime-ubuntu22.04
|
||||
# CUDA + cuDNN runtime so onnxruntime-gpu can use the card (it needs cuDNN 9 —
|
||||
# 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
|
||||
RUN apt-get update \
|
||||
|
||||
Reference in New Issue
Block a user