From 71154b7b4c466be38231c96e43c23415773cb694 Mon Sep 17 00:00:00 2001 From: henryruhs Date: Thu, 26 Sep 2024 09:38:54 +0200 Subject: [PATCH] Downgrade to CUDA 12.4 --- Dockerfile.cuda | 2 +- Dockerfile.tensorrt | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/Dockerfile.cuda b/Dockerfile.cuda index 7a29b5e..b7900d1 100644 --- a/Dockerfile.cuda +++ b/Dockerfile.cuda @@ -1,4 +1,4 @@ -FROM nvidia/cuda:12.6.1-cudnn-runtime-ubuntu22.04 +FROM nvidia/cuda:12.4.1-cudnn-runtime-ubuntu22.04 ARG FACEFUSION_VERSION=3.0.0 ENV GRADIO_SERVER_NAME=0.0.0.0 diff --git a/Dockerfile.tensorrt b/Dockerfile.tensorrt index 4081a4c..d069b7c 100644 --- a/Dockerfile.tensorrt +++ b/Dockerfile.tensorrt @@ -1,4 +1,4 @@ -FROM nvidia/cuda:12.6.1-cudnn-runtime-ubuntu22.04 +FROM nvidia/cuda:12.4.1-cudnn-runtime-ubuntu22.04 ARG FACEFUSION_VERSION=3.0.0 ENV GRADIO_SERVER_NAME=0.0.0.0