FloodDiffusion-Streaming

Runtime error

H-Liu1997 commited on Mar 16

Commit

3042363

1 Parent(s): c5fa989

fix: correct CUDA base image tag to 12.4.1

Files changed (1) hide show

Dockerfile CHANGED Viewed

@@ -1,4 +1,4 @@
-FROM nvidia/cuda:12.4.0-cudnn-runtime-ubuntu22.04
 RUN apt-get update && apt-get install -y \
     python3 python3-pip git \

+FROM nvidia/cuda:12.4.1-cudnn-runtime-ubuntu22.04
 RUN apt-get update && apt-get install -y \
     python3 python3-pip git \