Use cudnn8 instead of cudnn9 with nvidia cuda

LibrePhotos · Dec 22, 2024 · 2a61210 · 2a61210
1 parent 4411d59
commit 2a61210
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/backend-gpu/base/Dockerfile b/backend-gpu/base/Dockerfile
@@ -1,5 +1,5 @@
 # Stage 1: Build Stage
-FROM pytorch/pytorch:2.5.1-cuda11.8-cudnn9-devel AS builder
+FROM nvidia/cuda:11.8.0-cudnn8-devel-ubuntu22.04 AS builder
 ARG TARGETPLATFORM
 ENV DEBIAN_FRONTEND=noninteractive